本教程首先要安装并配置anaconda,如果不具备此条件,请先安装anaconda。
参考链接:Anaconda安装配置教程(Python)
1、查看本机的CUDA版本
cmd命令行输入nvidia-smi
,在第一行最右边可以看到CUDA的版本号。
2、安装Pytroch
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
文章出处登录后可见!
已经登录?立即刷新