显卡驱动对应cuda是12.4,但是这个没有对应的pytorch

安装cuda11.3,与pytorch 版本不兼容

cuda 建议安装11.8稳定版本

然后直接用官方命令暗转pytorch即可

conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

python=3.8

Logo

腾讯云面向开发者汇聚海量精品云计算使用和开发经验,营造开放的云计算技术生态圈。

更多推荐