torch.version.cuda在某些版本的pytorch和cuda中为None

ohfgkhjo  于 2023-01-09  发布在  其他
关注(0)|答案(1)|浏览(1240)

我的cuda版本显示在这里。nvcc -V outputnvidia-smi output
我基本上想安装apex。我首先使用命令

conda install pytorch=1.10.1 cudatoolkit=11.1 -c pytorch

用cuda安装torch,这个版本的cudatoolkit工作正常,

torch.version.cuda

正确显示预期输出“11.1”和

torch.cuda.is_available()

returns True.然而,当我稍后尝试安装apex时,这不起作用,因为上面显示的nvcc -V版本是11.4,而cudatoolkit是11.1。
然后我用命令

conda install pytorch cudatoolkit=11.4 -c pytorch

安装cudatoolkit的正确版本11.4。完成此操作后,torch版本变为1.9.1。但是,

torch.version.cuda

变为“无”并且

torch.cuda.is_available()

返回False。
我也在torch版本1.9.1和1.12.1中尝试过cudatoolkit=11.4,但它们都不起作用。
我希望我的GPU能被正确检测到。我应该使用什么版本的torch和cudatoolkit?或者这里还有什么问题吗?非常感谢。

7kqas0il

7kqas0il1#

我有同样的问题,使用pip安装解决了我的问题。

pip3 install torch==1.10.1+cu113 torchvision==0.11.2+cu113 torchaudio==0.10.1+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html

相关问题