使用集成显卡/集显(AMD)或者GPU安装运行pytorch?步骤解析~

1、查看本地电脑显卡配置

Windows+r,打开命令行窗口,输入taskmgr,打开任务管理器,可以查看显卡配置

我这里的是AMD集成显卡,它与处理器共享内存,并且通常用于处理计算机的基本图形和显示操作。与NVIDIA做深度学习相比逊色,但是也能使用pytorch。

2、打开pytorch官网:

https://pytorch.org/

进入官网后,往下翻

复制命令,来到终端,我这里使用的是pycharm终端。(其实把pycharm中settings---tools--- shell path-更改成cmd.exe,便和cmd打开的终端没有区别了,读者自行选择)

来到终端,输入之前复制的命令。(事先建议先创建一个虚拟环境,避免很多额外的bugs)创建虚拟环境的教程请看我之前的一篇文章:

https://blog.csdn.net/zhang___jian61/article/details/133711509

粘贴之后按enter键,之后输入y,确定之后继续等待。

安装完毕之后,继续在终端输入python,可以看到>>>表示进入了Python环境。

进入Python环境之后,检测是否安装成功,输入以下命令:

import torch

如果没有报错,就安装成功了。

但是如果继续输入torch.cuda.is_available(),检查当前系统是否支持CUDA函数,如果返回false,则可能是配置不当,或者显卡配置不够(笔者就是这种情况,有钱了升级配置~~~)

  • 3
    点赞
  • 17
    收藏
    觉得还不错? 一键收藏
  • 8
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 8
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值