2020-08-07

Win10搭建深度强化学习环境

现在课题涉及到深度强化学习这一块，因此需要在办公室的电脑上搭建DRL的环境，这两天摸爬滚打搭了出来，mark一下过程：
主要参考的教程有：
win10深度学习环境搭建
手把手安装Win10深度强化学习环境: Anaconda+TenserFlow+Pytorch+Gym
Win10下安装Anaconda+CUDA+cudnn+TensorFlow+keras+PyTorch+Pycharm

0.前言

笔者电脑基本配置：Win10；anaconda版本：4.7.12；虚拟环境dl内python版本：3.6.10；pycharm版本：2017.1。

1.安装CUDA

下载的是CUDA9.0版本，下载地址：https://developer.nvidia.com/cuda-toolkit-archive

2.安装cuDNN

下载的是7.5版本，下载地址：https://developer.nvidia.com/rdp/cudnn-archive
下载完成后，把文件夹里的3个文件分别放到CUDA安装目录下的对应位置（默认安装目录：C:\Program Files\NVIDIA GPU Computing Toolkit）

3.安装tensorflow

选择的是tensorflow的gpu1.11版本。要打开cmd，进入到anaconda3/Scripts目录下，输入命令activate dl（我的虚拟环境名称）进入到虚拟环境，之后退出该目录并进入到anaconda3/envs/dl目录下，输入：

pip install tensorflow_gpu==1.11

4.安装pytorch

这里安装的是torch-1.0.1-cp36的版本，下载地址可参考博客：https://blog.csdn.net/qq_27825451/article/details/90033909
之后和3中采用的方法在cmd命令中进入到dl目录，输入：

pip install xx\torch-1.0.1-cp36-cp36m-win_amd64.whl
pip install torchvision

这里XX是指你从博客里下下来的whl格式文件在你电脑里的存储位置。torchvision里有很多CV用的数据集，不过我安装不出来，会提示没有合适的版本，可能是scipy这边有问题，不过目前也用不到所以就没管了。