安装Pytorch(基于AMD显卡)

前提:安装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

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值