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:

 查看系统信息栏:

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

之后进入pytorch官网:

Start Locally | PyTorch

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

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

conda activate 环境名

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

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

        4. 安装gym

conda install gym

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值