CUDA11.7安装pytorch

根据官网,发现CUDA11.7并不在支持之中,最高支持CUDA11.6.Start Locally | PyTorch

CUDA11.7安装pytorch_第1张图片

 根据nvidia-smi查询得到我电脑是CUDA11.7,并不受支持。

CUDA11.7安装pytorch_第2张图片

 解决方案:安装CUDA11.3。我测试过用CUDA11.7的驱动安装pytorch-CUDA11.6是无法使用GPU的! 

正确命令:conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

运行结果如下:

 

你可能感兴趣的:(pytorch,人工智能,python)