前提:安装Anaconda,Python的库大部分都在Anaconda的Anaconda prompt中进行安装各种包(注意:Anaconda的位数要和Python保持一致)
1.创建环境
一般情况下所有的 Anaconda prompt中默认的都是base环境。
接下来需要再创建Pytorch环境,重新创建一个环境就相当于平时大家的安装内容都在C盘,C盘默认的内容很多,所以需要把适当的放到D盘,防止空间不够
conda create -n pytorch python=3.7
创建python版本为3.7的pytorch环境(python版本可以随意选择)
安装完之后会出现y/n选项,输入y即可
此时,pytorch环境就设置成功,但是!只有了环境,内容还是空的。激活并进入pytorch环境
进入之后,接下来需要安装pytorch包
2.安装Pytorch
amd显卡的需要使用cpu版本的pytorch语句安装,现在大部门是cuda语句,因为市面上好多电脑的显卡是NVIDIA(英伟达)
确认显卡是否为AMD显卡
接下来等待安装,安装过程比较长,请耐心等待
安装完成之后,输入pip list ,查看是否安装成功
3.检验是否安装成功
输入python ,会输出对应的信息,比如python版本、显卡版本、系统位数等
随后输入import torch 不出错就说明导入torch成功!!!
参考:
https://blog.csdn.net/CUFEECR/article/details/137015313
https://blog.csdn.net/Kingzhangfang/article/details/137331161