环境配置--解决torch.cuda.is_available()返回:False

在网上很多都说是因为pytorch版本和CUDA不匹配造成的,但对于我的情况并不是,而是因为在官网默认下载到了的pytorch是cpu版本(真坑!另外在pycharm默认下载到的pytorch == 1.9.0也是cpu版本,用不了cuda)

1. 首先查看CUDA是否安装成功:

在终端输入:

ncvv --version

 环境配置--解决torch.cuda.is_available()返回:False_第1张图片

 有CUDA版本返回,证明是安装成功的;

2.查看pytorch是否安装成功

在终端输入python进入python环境:

python

 然后import 一下,并返回当前的版本号:

import torch

torch.__version__

 

证明pytorch也安装成功了,并确认cuda版本和pytorch兼容!(但是不知道这个torch是cpu版本还是cuda版本,在pycharm输入print(torch.__version__),如果返回结果后缀有-cpu那证明是版本错误) 

3.更换pytorch版本:

进入链接:https://download.pytorch.org/whl/torch_stable.html

直接点击下载CU100+ + win/linux(cu表示cuda版本,且版本号为10.0,以此类推)

下载完成后,在终端使用指令进行安装pytorch (后缀名也不能少!):

pip install cu100/torch-1.1.0-cp36-cp36m-win_amd64.whl

你可能感兴趣的:(BUG!)