Pytorch安装教程(Python)

本教程首先要安装并配置anaconda,如果不具备此条件,请先安装anaconda。
参考链接:Anaconda安装配置教程(Python)

1、查看本机的CUDA版本

cmd命令行输入nvidia-smi,在第一行最右边可以看到CUDA的版本号。

 2、安装Pytroch

1、点击进入Pytorch官网

 2、下拉至下图位置,只需保证系统和电脑匹配,CUDA版本小于本机CUDA版本,调整结束后,复制红框中的指令

 3、打开Anaconda Prompt,粘贴指令,回车执行

 4、输入y确认,回车

5、等待安装直到左上角出现done

6、输入pip list,查看安装了哪些包

 

7、输入python进入环境,输入import torch,无报错代表导入成功(这里要等待一段时间)

 8、输入torch.cuda.is_available()查看torch是否可以使用显卡,True代表可以

9、正常情况到这里,pytorch的环境已经安装好了,如果发现Pycharm中仍报错找不到包,说明是 pycharm中的解释器没有改成anaconda中的python.exe。

解决办法参考:http://t.csdn.cn/T1ptC

文章出处登录后可见!

已经登录?立即刷新

共计人评分,平均

到目前为止还没有投票!成为第一位评论此文章。

(0)
心中带点小风骚的头像心中带点小风骚普通用户
上一篇 2023年11月9日
下一篇 2023年11月9日

相关推荐