windows 安装pytorch和gym环境

        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:

windows 安装pytorch和gym环境_第1张图片

 查看系统信息栏:

windows 安装pytorch和gym环境_第2张图片

 查看CUDA一栏的驱动版本:11.4.161

之后进入pytorch官网:

Start Locally | PyTorch

windows 安装pytorch和gym环境_第3张图片

选择比驱动版本号低的选项即可:比如这里选择了CUDA 11.3 这个选项

之后切换到要安装pytorch的那个环境:

conda activate 环境名

 将安装命令复制进终端中执行即可

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

        4. 安装gym

conda install gym

你可能感兴趣的:(pytorch,深度学习)