windows环境下anaconda+pycharm+cuda+cudnn+pytorch安装踩坑大全
- anaconda安装
- pycharm安装
- cuda
- cudnn
- pytorch
- gym环境安装
anaconda安装
安装时不要勾选添加环境变量,完事后自己去手动添加。添加时注意是用户变量还是系统变量,我这个就搞错了,导致重装了一遍。
另外就是anaconda换源,比较简单。同时把pip源也给换了吧,就是在windows的c盘目录下写一个小文件即可,比较简单。
再就是创建虚拟环境,不必命令行式的创建,打开图形化操作界面,创建更简单。
在虚拟环境里如何安装第三方库,只需在相应的虚拟环境里,启动终端,pip install 即可,这里承接上面的换源步骤。
pycharm安装
比较简单,基本没坑
cuda
安装卸载重装了三次才搞定
首先是版本号,去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,整个过程比较费时,跟着教程慢慢来即可。
cudnn
从nvidia官网,下载与cuda版本号相对应的cudnn,这个文件比较小,是压缩文件。
下载完之后,把压缩包里面的文件夹里面的文件,拷贝到cuda安装目录下的同名文件夹里。注意是文件的复制与粘贴,而不是文件夹的复制与替换,我就是因为这个问题重装了几次。
pytorch
torch包也比较大,两三个G。推荐使用本地whl文件的方式安装,从官网下载对应于cuda版本的、gpu版的、对应于你的Python语言版本的(例如3.6)torch文件和torchversion两个文件。
下载好之后,打开python虚拟环境下的终端,cd到torch文件和torchversion两个文件的位置,pip install 这两个文件即可。注意,需要pip换源。
gym环境安装
在虚拟环境的命令行里输入 pip install gym[all]即可