一.Pytorch的版本选择

1.打开官网(官方网址:PyTorch),下载时看看下载的CUDA再选择(没下CUDA的可以看博主以前的blog)
 

                                                         (看好CUDA的版本再下载)

                       

                                              (CUDA版本老也可以看以前的Pytorch版本)

2.复制下载路径

二.在Anaconda Prompt安装Pytorch

1.打开Anaconda Prompt(没安装Anaconda的可以看博主以前的blog)

2.创建一个虚拟环境(输入conda create -n pytorch python=3.9)(python不需要太高版本,但也不要低于3.9)

                                                                 (选择y(就是yes))  

3.建立好叫pytorch的虚拟环境 

(此时输入conda activate pytorch)

                               

                                                   (已经是Pytorch虚拟环境)

4.将之前官网复制的下载路径粘贴过来

(推荐白天下载)

三.检验Pytorch是否安装成功

 在pytorch虚拟环境下输入python

再输入import torch(Enter后)输入torch.cuda.is_available()

如果出现True表明安装成功

 

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐