两步搞定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好像不需要另外安装,估计是打包一起安装了
点个赞吧!
标签:11.1,pytorch,conda,cuda11.1,官网,安装,RTX3090 来源: https://blog.csdn.net/SUPER_ONE1/article/details/120513160