栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

windows 安装pytorch和gym环境

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

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

暑期编程PK赛 得CSDN机械键盘等精美礼品!
转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/1015690.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号