一.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

DAMO开发者矩阵,由阿里巴巴达摩院和中国互联网协会联合发起,致力于探讨最前沿的技术趋势与应用成果,搭建高质量的交流与分享平台,推动技术创新与产业应用链接,围绕“人工智能与新型计算”构建开放共享的开发者生态。

更多推荐