首页 > 编程学习 > Pytorch 亲妈级安装教程(GPU or CPU版本)

Pytorch 亲妈级安装教程(GPU or CPU版本)

发布时间:2022/11/10 2:12:00

本文只讲干货。

首先就是咱们需要确定下CPU还是GPU版本,那毋庸置疑,都玩这玩意了,CPU版本dog都不下!(开玩笑呢哈哈哈,如果你想下CPU版的也行,安装过程也简单)

上来就是版本对应问题,版本对应问题可以去看我的tensorflow安装的文章(这篇文章里的很多东西这篇文章可以用到):

Tensorflow亲妈级安装教程(CPU和GPU版)_深海鱼肝油ya的博客-CSDN博客

pytorch这玩意安装时自带cuda和cudnn(这可比tensorflow安装时方便多了),我要是在anaconda的虚拟环境里安装的话,那这个的版本选择应该还会受限于电脑显卡驱动版本的影响吧,我的cuda和cudnn是本科的时候装的,有点老了,10.0的,我看官网,cuda10.0对应pytorch是1.2.0,现在都1.13了,差的有点多;不过因为pytorch自带cuda啥的,而且咱们是在conda的虚拟环境里安装,这就方便太多,我的显卡驱动版本能对应到的最高的cuda版本是11.0,我看官网上cuda10.2 都能下pytorch1.12.1的版本,还是可以的,蛮舒服。

接下来就是anaconda创建虚拟环境,然后激活(进入)环境

然后去pytorch官网:PyTorch

在主页这里选择配置完毕之后,在最底下会生成安装命令,复制之后到conda虚拟环境下(命令行形式)粘贴执行即可(记得把-c pytorch去掉),执行命令之前可以先为conda设置清华源,参考下面这篇文章:

conda配置清华源_blueman8888的博客-CSDN博客_conda配置清华源

用户目录就是像下面那样,当然不同人的电脑可能文件名称不一样。

换了源之后就可执行命令,等待安装成功!

解决pytorch安装下载慢(清华源+缓存到本地):

亲测绝对有效——解决pytorch安装下载慢(清华源+缓存到本地) - 灰信网(软件开发博客聚合)

安装过程也可以参考下面两篇文章:

Windows下安装pytorch-GPU版本(小白踩坑安装过程记录)_一只爱学习的兔子的博客-CSDN博客

PyTorch安装教程_诸神缄默不语的博客-CSDN博客_pythorch安装


安装成功之后检测:

import  torch 
#检测使用了哪个GPU
print(torch.cuda.get_device_name(0))
#torch版本
print(torch.__version__)
#cuda的版本
print(torch.version.cuda)

Copyright © 2010-2022 dgrt.cn 版权所有 |关于我们| 联系方式