0.问题出现背景
ps: win10 64位环境。cuda已经安装,显卡驱动版本472对应torch1.8.0版本对应cuda10.2版本都没问题。
因为硬盘坏了,所以从头开始装按照深度学习系列(一)从零相关软件安装篇(笔记)_lry320的博客-CSDN博客
安装最新的发现,最后
>>>print('gpu',torch.cuda.is_available())
gpu False
后来发现可能是因为我的Anaconda改了清华源,虽然用的官网Previous PyTorch Versions | PyTorch
的命令下载torch1.8带GPU的命令,但是却只能下载cpu的版本。
1.解决方法
a.去官网Start Locally | PyTorch选择pip使用网站下载
pip3 install torch==1.8.2+cu102 torchvision==0.9.2+cu102 torchaudio===0.8.2 -f https://download.pytorch.org/whl/lts/1.8/torch_lts.html
运行结果如下:
>>> import torch
>>> print(torch.__version__)
1.8.2+cu102
>>> print('gpu',torch.cuda.is_available())
gpu True
b.离线环境下通过Anaconda安装相关包
手动去Index of /anaconda/cloud/pytorch/win-64/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror
找到自己对应的版本下载。 此处我选择了最新的pytorch-1.9.1-py3.9_cuda10.2_cudnn7_0.tar.bz2
版本对应表格https://github.com/pytorch/vision#installation,最新下载的Anaconda自带的Python3.9版本。注意安装的torchvison版本即可。
下载好压缩包后,用cmd进入解压的文件夹中(直接在当前文件夹位置地址栏输入%comspec%,进入该目录的命令提示符),运行命令
conda install pytorch-1.9.1-py3.9_cuda10.2_cudnn7_0.tar.bz2