当电脑gpu太强,安装最新对应的pytorch版本和对应版本的cuda
当电脑gpu太强,安装最新对应的pytorch版本和对应版本的cuda
·
首先我的电脑的gpu为rtx4060,对应的cuda版本为12.3,而pytorch最新的cuda对应的版本为12.1,所以先要装上12.1的cuda


我们去cuda官网找到12.1版本的驱动:https://developer.nvidia.com/cuda-toolkit-archive
选择12.1的cuda版本点进去
对应的选择windows11版本exe(local),然后进行本地安装,安装之前进行本地的cuda删除,删到最后只保留三个。

直接一直下一步就能安装好,下面这样就安装好了。
安装最新的annaconda,可以参考那篇安装annaconda的博客
将对应的pytorch安装上
conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia
出现下面情况就是安装成功
安装最新的pycharm
先去安装一个最新的专业版的pycharm,然后找淘宝破解一下,安装的时候一直点下一步,主要是下面这个全部选上
最后,在pycharm中进行环境切换,切到下面这样就可以了,以前的pycharm需要一个一个的添加环境,这次新版本只要添加scripts下的conda可执行文件,下面就会出来你创建的所有环境
欢迎来到FlagOS开发社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。
更多推荐
所有评论(0)