其他分享
首页 > 其他分享> > 两步搞定RTX3090+cuda11.1+pytorch环境配置

两步搞定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