windows下 anaconda + cuda + pytorch 安装后gpu不可用问题
本记录使用的软件版本截图:
一、关键问题
1、cuda版本
cuda 使用
2、NVIDA控制面板查看本机显卡驱动版本
二、注意事项
CUDA版本选择 不能选择None ,none表示不使用GPU
***** 由于开始使用的是ancaonda安装的pytorch , 导致 pycharm中
torch.cuda.is_available() 返回值为 false。 因为anaconda安装的pytorch 版本不带gpu。
解决办法:
以管理员身份打开anaconda ,卸载pytorch。 并打开pytorch 官网,
复制 上图最下面的命令,到anaconda prompt 命令行。 安装带GPU版本的pytorch。
至此 ,则应该可以使用 GPU。
验证方法:
打开cmd命令行,进行如下输入:
感谢网上不知名的网友的热心帮助。 加油咯