赞
踩
根据官网,发现CUDA11.7并不在支持之中,最高支持CUDA11.6.Start Locally | PyTorch
根据nvidia-smi查询得到我电脑是CUDA11.7,并不受支持。
解决方案:安装CUDA11.3。我测试过用CUDA11.7的驱动安装pytorch-CUDA11.6是无法使用GPU的!
正确命令:conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
运行结果如下:
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。