我的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?或者这里还有什么问题吗?非常感谢。
1条答案
按热度按时间7kqas0il1#
我有同样的问题,使用pip安装解决了我的问题。