环境配置--解决torch.cuda.is_available()返回:False
在网上很多都说是因为pytorch版本和CUDA不匹配造成的,但对于我的情况并不是,而是因为在官网默认下载到了的pytorch是cpu版本(真坑!另外在pycharm默认下载到的pytorch == 1.9.0也是cpu版本,用不了cuda)
1. 首先查看CUDA是否安装成功:
在终端输入:
ncvv --version
有CUDA版本返回,证明是安装成功的;
2.查看pytorch是否安装成功
在终端输入python进入python环境:
python
然后import...
原创
2021-07-30 16:26:11 ·
1577 阅读 ·
0 评论