省流版:
终端中配置好的torch不能在虚拟环境中直接使用,需要进入虚拟环境中检查配置。

在终端中安装好对应版本的torch后在conda虚拟环境中仍返回False

笔者这里首先在终端中安装了torch
在这里插入图片描述
但是在虚拟环境py38中进行测试时仍旧报错
在这里插入图片描述
这里笔者的理解是虚拟环境中的torch和本机中的torch是隔离的,所以需要检查替换虚拟环境中的torch版本。

首先检测torch版本,因为直接进行pip安装的torch可能是cpu版本

import torch
print(torch.__version__)

在这里插入图片描述
此时可以看到虚拟环境中的torch版本就错了,先卸载

 pip uninstall torch

在这里插入图片描述

重新在虚拟环境中安装对应版本的torch

如何找到正确版本我这里就不再赘述。

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

在这里插入图片描述

再次进行测试

在这里插入图片描述
此时版本正确且 torch.cuda.is_available()返回True。
跑一下代码也是ok的。
在这里插入图片描述

Logo

腾讯云面向开发者汇聚海量精品云计算使用和开发经验,营造开放的云计算技术生态圈。

更多推荐