试了很多方法,查了很多教程,总结的经验
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环境配置的全部内容,更多相关RTX3090+cuda11内容请搜索靠谱客的其他文章。
本图文内容来源于网友提供,作为学习参考使用,或来自网络收集整理,版权属于原作者所有。
发表评论 取消回复