pytorch安装及使用GPU失败的解决办法 |
您所在的位置:网站首页 › muta为什么用不了 › pytorch安装及使用GPU失败的解决办法 |
先打开cmd,输入nvidia-smi查看本电脑最大能支持的cuda版本号: 去pytorch官网:PyTorch下载低于本电脑cuda号的。 如果使用CPU训练,Compute Platform选择CPU。 在虚拟环境中运行刚才在pytorch官网生成的下载指令。 |
今日新闻 |
点击排行 |
|
推荐新闻 |
图片新闻 |
|
专题文章 |
CopyRight 2018-2019 实验室设备网 版权所有 win10的实时保护怎么永久关闭 |