admin管理员组

文章数量:1663032

核心:在显卡安装的cuda版本适配的pytorch中,更换pytorch的版本

刚遇到这个错误时,在网上搜索了一下,感谢博主1和博主2的解决方法带给我的启发。

标题服务器cuda是11.3版本,配置其他环境“御用”的pytorch安装语句

conda install pytorch1.12.1 torchvision0.13.1 torchaudio==0.12.1 cudatoolkit=11.3 -c pytorch

但就是使用上述语句安装的pytorch版本会和待调试的代码有冲突

更换为pytorch==1.11.0解决问题!

conda install pytorch1.11.0 torchvision0.12.0 torchaudio==0.11.0 cudatoolkit=11.3 -c pytorch

上面语句来自pytorch官网

本文标签: InvalidErrornvrtcArchArchitecture