在conda中安装PyTorch | pkgs中已有gpu pytorch的安装包,但仍无法使用gpu pytorch

今天跑模型时,发现自己之前安装的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 。

你可能感兴趣的:(深度学习,Debug,anaconda,gpu,cuda,深度学习,pytorch)