所以不久前,我使用pip安装了pytorch,并将其作为计算平台"仅CPU";。现在我想在GPU上使用torch,但无论我如何尝试将CUDA作为计算平台安装,它仍然总是使用1.9.0+cpu版本。pip和conda都无法卸载它,我甚至试图手动删除一些torch文件,但没有任何帮助。无论我怎么尝试,它仍然总是使用CPU计算设备运行,而不是使用CUDA支持。我还尝试创建不同的虚拟环境并在那里安装CUDA,但它仍然适用于CPU版本。
当我尝试用pip 卸载时
当我尝试用conda 卸载时
import torch
print(torch.__version__)
这个打印:
在此处输入图像描述
PS:我很抱歉这些图片是链接,我没有足够的声誉为实际的图片显然
要让我的评论成为一个你可以接受的答案:
你可以使用
print(torch.__file__)
查看神秘的Torch安装在哪里并将其删除。
如果这不起作用,你可以随时在电脑上搜索torch
。