(1)问题:在CUDA,cudnn,已装好,指令
conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia
一直转圈,不得已使用pip指令执行完成之后,torch.cuda.is_available()依旧返回False
pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117
(2)解决办法:
先 从https://download.pytorch.org/whl/torch_stable.html找到三个文件:
因为创建的环境(环境名为pytorch)里Python选择的3.9版本,所以这三个文件也是3.9版本
①torch-1.8.0+cu101-cp39-cp39-win_amd64.whl (cp39指的是Python3.9版本,win_amd64指的是64位Windows操作系统)(重新编辑内容:这里的cu101貌似是cudnn的版本即10.1版本的cudnn)
②torchvision-0.9.0+cu101-cp39-cp39-win_amd64.whl
③torchaudio-0.8.0-cp39-none-win_amd64.whl
以上三个文件,并不是唯一一组,只要对应即可 ,发现下载速度慢可以使用迅雷下载,下载完成后将这三个文件复制到下面文件中(非必要,当时想的要是安装肯定要复制到包下面的嘛,然后后面好像没有用到):
之后,先进入cmd,将三个文件使用pip install+文件名安装,注意,必须先安装torch-1.8.0+cu101-cp39-cp39-win_amd64.whl且这三个文件必须在当前目录下
pip install torch-1.8.0+cu101-cp39-cp39-win_amd64.whl pip install torchvision-0.9.0+cu101-cp39-cp39-win_amd64.whl
pip install torchaudio-0.8.0-cp39-none-win_amd64.whl
此处测试torch.cuda.is_available() 返回已经是true,但anaconda prompt里面仍然是false,
再然后进入anaconda prompt,命令进入conda activate pytorch再次安装三个文件:
测试,返回true,成功解决。
在这篇文章中,所能解决的只是一小部分返回false的问题。