Pytorch安装教程(Python)

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

1、查看本机的CUDA版本

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

Pytorch安装教程(Python)_第1张图片

 2、安装Pytroch

1、点击进入Pytorch官网

Pytorch安装教程(Python)_第2张图片

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

Pytorch安装教程(Python)_第3张图片

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

 4、输入y确认,回车

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

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

 Pytorch安装教程(Python)_第4张图片

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

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

Pytorch安装教程(Python)_第5张图片

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

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

你可能感兴趣的:(Python,python,pytorch,人工智能,conda)