1.Anaconda安装
Anaconda是python语言下的包管理软件,集成了较多功能。用python编程时推荐安装它
Anaconda | The World's Most Popular Data Science Platform
2. 创建新的python环境
为了更好的进行版本管理,防止可能发生的冲突,需要新建一个conda环境,之后的pytorch和gym包安装在这个环境下即可。
conda create -n 环境名 python=3.8
在 anaconda prompt 终端内输入以上命令,就会在默认的环境安装路径下安装一个新的环境,这个环境下的python版本是3.8
要了解conda create命令的语句格式,使用-h 选项
conda create -h
为了避免环境都安装到了C盘里,需要修改一下新环境的安装位置,有下面2种方法:
2.1 在命令里指明安装路径:
conda create --prefix=PATH python=3.8
这里的PATH是绝对路径
2.2 另一种一劳永逸的方法是修改默认环境安装位置:
首先查看配置的详细信息
conda config --show
找到envs_dirs 其中的第一条路径就是默认安装路径。如果要修改它,使用下面命令把它覆盖掉:
conda config --add envs_dirs PATH
PATH 这里写要安装的路径
3. 安装pytorch
如果要使用CUDA加速,就要选择合适的版本
首先要查看自己电脑上的CUDA版本:
打开NVIDIA Control Panel:
查看系统信息栏:
查看CUDA一栏的驱动版本:11.4.161
之后进入pytorch官网:
选择比驱动版本号低的选项即可:比如这里选择了CUDA 11.3 这个选项
之后切换到要安装pytorch的那个环境:
conda activate 环境名
将安装命令复制进终端中执行即可
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
4. 安装gym
conda install gym