今天跑模型时,发现自己之前安装的GPU pytorch用不了了,版本是1.7.0的。
>>> import torch
>>> torch.cuda.is_available()
False
不知道是什么原因。仔细回忆了一下,可能是因为之前自己 pip install 了 torchvision 工具包的缘故。
看了一下conda中存放python库的目录pkgs,发现有pytorch-1.8.0-py3.8_cuda10.2_cudnn7_0
这么一个包的存在,这不就是GPU pytorch包吗?看了下这个文件的大小有2GB。那应该是安装了gpu pytorch的啊。
后来请教了一下某宝的专业人士,他的意思是因为我下载了cpu pytorch,把原来的gpu pytorch给盖掉了,只要卸载cpu pytorch即可。于是在anaconda prompt运行pip uninstall torch
后,结果连import torch
也开始报错了,找不到该包了…
后来想起自己之前是手动下载了1.7.0版本的gpu pytorch,安装包还留着。于是就重新安装了一下安装包。就是下面这个包。
运行
pip install 路径\torch-1.7.0+cu110-cp38-cp38-win_amd64.whl
安装完了以后,
>>> import torch
>>> torch.cuda.is_available()
True
于此,成功重装了gpu pytorch 。