anaconda+cuda+cudnn+pytorch安装踩坑大全

安装时不要勾选添加环境变量,完事后自己去手动添加。添加时注意是用户变量还是系统变量,我这个就搞错了,导致重装了一遍。

另外就是anaconda换源,比较简单。同时把pip源也给换了吧,就是在windows的c盘目录下写一个小文件即可,比较简单。

再就是创建虚拟环境,不必命令行式的创建,打开图形化操作界面,创建更简单。

在虚拟环境里如何安装第三方库,只需在相应的虚拟环境里,启动终端,pip install 即可,这里承接上面的换源步骤。

比较简单,基本没坑

安装卸载重装了三次才搞定

首先是版本号,去Pytorch官网查询pytorch所支持的cuda版本号。有的cuda版本pytorch已经不支持了。例如”CUDA-10.2 PyTorch builds are no longer available for Windows, please use CUDA-11.3″另外你还要查一下自己电脑显卡所支持的最高cuda版本号,你只可以下载这个最高及最高以下版本的cuda。

其次就是下载了,cuda比较大,两三个G,稍安勿躁,慢慢下载即可。

再就是安装cuda,整个过程比较费时,跟着教程慢慢来即可。

从nvidia官网,下载与cuda版本号相对应的cudnn,这个文件比较小,是压缩文件。

下载完之后,把压缩包里面的文件夹里面的文件,拷贝到cuda安装目录下的同名文件夹里。注意是文件的复制与粘贴,而不是文件夹的复制与替换,我就是因为这个问题重装了几次。

torch包也比较大,两三个G。推荐使用本地whl文件的方式安装,从官网下载对应于cuda版本的、gpu版的、对应于你的Python语言版本的(例如3.6)torch文件和torchversion两个文件。

下载好之后,打开python虚拟环境下的终端,cd到torch文件和torchversion两个文件的位置,pip install 这两个文件即可。注意,需要pip换源。

在虚拟环境的命令行里输入 pip install gym[all]即可

Original: https://blog.csdn.net/xiao_sir/article/details/123535851
Author: 小橘龙
Title: anaconda+cuda+cudnn+pytorch安装踩坑大全

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/709603/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球