本文主要是介绍RTX3090+cuda11.1+pytorch环境配置,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
试了很多方法,查了很多教程,总结的经验
3090显卡只支持cuda11以上的版本,不要再装cuda10了!!!不然白搞
安装流程:
1.去pytorch官网,一定要去官网!!!看官网支持那个cuda版本,目前看支持11.1版本,那就老老实实的安装cuda11.1,别整些有的没的。
2. 然后就在自己的环境里按照官网给的命令安装
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c conda-forge
cudnn好像不需要另外安装,估计是打包一起安装了
点个赞吧!
这篇关于RTX3090+cuda11.1+pytorch环境配置的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!