环境配置--解决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 ·
1505 阅读 ·
0 评论