如何搭建pytorch环境的方法步骤

站长资源 2024/11/24 佚名

2 0 1

1.conda创建虚拟环境pytorch_gpu

conda create -n pytorch_gpu python=3.6

创建虚拟环境还是相对较快的，它会自动为本环境安装一些基本的库，等待时间无需很长，成功之后界面如下所示：

2.切换到pytorch环境

使用如下命令，切换到我们刚刚创建好的pytorch虚拟环境，这样我们避免与其它python环境之间的干扰。

conda activeta pytorch_gpu

切换成功之后就会看到在路径前边显示我们已经进入该虚拟环境。

3.安装几个常用库（也可暂时不安）

conda install pandas jupyter notebook

4.安装pytorch

4.1进入官网查看要下载的版本

查看对应的版本，这里是官方链接：

4.2 根据系统信息及cuda版本选择对应toolkit

这里最主要的是那个CUDA的版本，此处我选择的是10.1，是因为我的电脑的 cuda版本信息就是这样的。具体的查看方法可在4.5节查阅。

4.3复制上图中最后一行代码到pytorch环境终端

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

这里是下载过程截图：

但是在下载过程中torchvision以及pytorch没有下载成功，

因为这是去其官网下载，下载速度很慢，在上图中我们也可以看出是因为网络错误,网络上也有几种其他的方法，此处我没去验证，我还是让电脑重新下载的，等待时间挺长的，但是因为是在晚上下载的，一早起来就好了

4.4 验证pytorch是否安装成功

此时直接输入 python，即可成功进入：

而后输入如下指令，查看torch是否安装成功

> import torch
> x=torch.randn(4,4)
> print(x)

正常情况下是出现这个界面的：

验证完成之后，可以quit()保存退出。

4.5 如何查看自己电脑cuda版本

4.5.1 windows如何查看

NVDIA控制面板–>帮助–>系统信息

组件–>NVCUDA.DLL 可以查看CUDA版本

这里我的显示是10.1,所以我上边下载的版本也是10.1的，这里的版本要对应上，否则会出现问题。

4.5.2 linux如何查看

打开终端，输入：nvcc -V

nvcc -V

或者如下方式查看：

CUDA:

cat /usr/local/cuda/version.txt

cudnn:

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

pytorch搭建环境,pytorch搭建

广告合作：本站广告合作请联系QQ：858582 申请时备注：广告合作（否则不回）
免责声明：本站资源来自互联网收集,仅供用于学习和交流,请遵循相关法律法规,本站一切资源不代表本站立场,如有侵权、后门、不妥请联系本站删除！

评论“如何搭建pytorch环境的方法步骤”

暂无评论...

《魔兽世界》大逃杀！60人新游玩模式《强袭风暴》3月21日上线

暴雪近日发布了《魔兽世界》10.2.6 更新内容，新游玩模式《强袭风暴》即将于3月21 日在亚服上线，届时玩家将前往阿拉希高地展开一场 60 人大逃杀对战。

艾泽拉斯的冒险者已经征服了艾泽拉斯的大地及遥远的彼岸。他们在对抗世界上最致命的敌人时展现出过人的手腕，并且成功阻止终结宇宙等级的威胁。当他们在为即将于《魔兽世界》资料片《地心之战》中来袭的萨拉塔斯势力做战斗准备时，他们还需要在熟悉的阿拉希高地面对一个全新的敌人──那就是彼此。在《巨龙崛起》10.2.6 更新的《强袭风暴》中，玩家将会进入一个全新的海盗主题大逃杀式限时活动，其中包含极高的风险和史诗级的奖励。
《强袭风暴》不是普通的战场，作为一个独立于主游戏之外的活动，玩家可以用大逃杀的风格来体验《魔兽世界》，不分职业、不分装备（除了你在赛局中捡到的），光是技巧和战略的强弱之分就能决定出谁才是能坚持到最后的赢家。本次活动将会开放单人和双人模式，玩家在加入海盗主题的预赛大厅区域前，可以从强袭风暴角色画面新增好友。游玩游戏将可以累计名望轨迹，《巨龙崛起》和《魔兽世界：巫妖王之怒经典版》的玩家都可以获得奖励。

更新日志

2024年11月24日

如何搭建pytorch环境的方法步骤

Python3自动生成MySQL数据字典的markdown文本的实现

Python基于Tkinter编写crc校验工具

评论“如何搭建pytorch环境的方法步骤”

《魔兽世界》大逃杀！60人新游玩模式《强袭风暴》3月21日上线

更新日志

友情链接