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

您所在的位置:网站首页 显卡没有cuda 使用集成显卡/集显(AMD)或者GPU安装运行pytorch?步骤解析~

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

2024-04-05 16:06| 来源: 网络整理| 查看: 265

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,则可能是配置不当,或者显卡配置不够(笔者就是这种情况,有钱了升级配置~~~)



【本文地址】


今日新闻


推荐新闻


CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3