时间戳

首页 > 解决方案 > 我可以使用高版本的torch和低版本的cuda吗？

pytorch - 我可以使用高版本的torch和低版本的cuda吗？

问题描述

我正在使用远程 GPU 设置我的 Conda 环境以使用 Pytorch。我用的GPU是only NVIDIA-SMI 396.54，所以只能用cuda 9.2版

但是，我需要使用更高版本的手电筒才能使用某些属性。

我试过了

conda install pytorch==1.7.1 torchvision==0.8.2 cudatoolkit=9.2

但这导致

print(torch.version.cuda)>> None

torch.cuda.is_available() >> False

标签： pytorch

解决方案

我要检查两件事。

在运行上述命令之前，您可能无意中安装了 pytorch cpu 版本或将其安装在您的环境中。即使你安装了 gpu 版本的 Pytorch，如果你已经有 pytorch 的 cpu 版本，那么torch.cuda.is_available()将返回 False。因此我建议查看这个链接： Forum on why Pytorch is CPU version even after installed cudatoolkit version

虽然，我很确定上面的事情是你的问题，但我建议看看第二件事。

要了解如何下载先前版本的 Pytorch，请参阅此链接。https://pytorch.org/get-started/previous-versions/

看了这个之后，我建议开始一个新的 conda env 并首先运行你的 conda install 命令。

萨塔克耆那教

推荐阅读