一、解决方案
通过 pytorch 官网下载 pytorch1.12 的稳定版本 对应的CUDA 11.3 的版本就可以。直接按照官网命令安装就可以,亲测可用。
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
参考链接
[1] 盖盖开心果. CUDA版本11.4,pytorch应该下哪个版本的?https://www.zhihu.com/question/482335084, 2021-12-19/2022-10-18.
版权声明:本文为qq_39779233原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。