Pytorch(一) Pytorch 的安装[通俗易懂]

Pytorch(一) Pytorch 的安装[通俗易懂]安装步骤1.检查是否有合适的GPU,若有安装Cuda与CuDNN2.下载pytorch安装文件,进行pytorch的安装检查电脑是否有合适的GPU在桌面上右击如果能找到NVIDA控制面板,则说明该电脑有GPU。控制面板如下,并通过查看系统信息获取支持的Cuda版本点击帮助->点击系统信息弹出下面的对话框,在驱动程序版本那一栏就能看到该计算机使用的驱动版本。…

大家好,又见面了,我是你们的朋友全栈君。

安装步骤

1.检查是否有合适的GPU, 若有安装Cuda与CuDNN

2.下载pytorch安装文件,进行pytorch的安装

检查电脑是否有合适的GPU

在桌面上右击如果能找到NVIDA控制面板,则说明该电脑有GPU。控制面板如下,并通过查看系统信息获取支持的Cuda版本

Pytorch(一) Pytorch 的安装[通俗易懂]

点击 帮助->点击 系统信息 弹出下面的对话框,在驱动程序版本那一栏就能看到该计算机使用的驱动版本。

Pytorch(一) Pytorch 的安装[通俗易懂]

 

下载Cuda

官网:https://developer.nvidia.com/cuda-10.1-download-archive-update2

https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html 这里可以查询到我们应该下载哪个版本

Pytorch(一) Pytorch 的安装[通俗易懂]

然后,根据实际情况选择合适的版本

Pytorch(一) Pytorch 的安装[通俗易懂]

下载CuDNN 

官网 https://developer.nvidia.com/rdp/cudnn-download

 

安装Cuda

1.与安装其他的软件类似

2.安装结束后将  ~/nvcc/bin(因为版本的不同可能在不同的地方) 目录添加到环境变量

3.在命令行下输入 nvcc -V, 出现下列信息说明Cuda安装成功

Pytorch(一) Pytorch 的安装[通俗易懂]

4.将CuDNN压缩包解压后,下面的三个文件夹复制到Cuda的安装目录下

Pytorch(一) Pytorch 的安装[通俗易懂]

5.然后执行Demo, 如果Demo中显示PASS则说明安装成功

Pytorch(一) Pytorch 的安装[通俗易懂]

 

安装Pytorch

官网 https://download.pytorch.org/whl/torch_stable.html 选择合适的版本

torch/torchvision 都需要安装

Pytorch(一) Pytorch 的安装[通俗易懂]

Pytorch(一) Pytorch 的安装[通俗易懂]

Pytorch(一) Pytorch 的安装[通俗易懂]

 

验证Pytorch

Pytorch(一) Pytorch 的安装[通俗易懂]

 

安装成功

通过上面输出的信息,说明安装是成功的

版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 举报,一经查实,本站将立刻删除。

发布者:全栈程序员-用户IM,转载请注明出处:https://javaforall.cn/152754.html原文链接:https://javaforall.cn

【正版授权,激活自己账号】: Jetbrains全家桶Ide使用,1年售后保障,每天仅需1毛

【官方授权 正版激活】: 官方授权 正版激活 支持Jetbrains家族下所有IDE 使用个人JB账号...

(0)


相关推荐

发表回复

您的电子邮箱地址不会被公开。

关注全栈程序员社区公众号