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

pytorch-gpu 安装 so-easy

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

pytorch-gpu 安装 so-easy

1.安装anaconda

pytorch库以及cuda的安装只需要在conda环境下一条指令即可

2.CUDA的安装与下载

这里可以参照我之前安装tensorflow的有两种方式,不建议去英伟达官网下载安装

3 pytorch库的安装


先创建pytorch环境并激活环境,环境名就叫“pytorch”吧。打开cmd命令提示符,输入:

conda create –n pytorch python=3.8
activate pytorch


成功激活环境后,可以看到命令提示符前会带有环境名,以后每次都要激活一下进入这个环境才能进行安装库等操作。
那么就可以通过访问这个网址查询对应pytorch版本conda安装指令

https://pytorch.org/get-started/previous-versions/

或者也可以访问这个地址 https://pytorch.org/get-started/locally/获得下载指令

在pytorch环境中直接输入下面这一行命令来安装pytorch

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

在cmd激活环境中输入以上指令即可

 

 

4.方法2

上一个方法对网速要求比较高,我们还可以选择另一个方法,分别安装好cuda和cuDNN后

所以,我们可以打开下面网站手动下载库:
https://download.pytorch.org/whl/torch_stable.html
我们选择
cu110/torch-1.7.1%2Bcu110-cp38-cp38-win_amd64.whl
下载地址:https://download.pytorch.org/whl/cu110/torch-1.7.1%2Bcu110-cp38-cp38-win_amd64.whl

再选择cu110/torchvision-0.8.2%2Bcu110-cp38-cp38-win_amd64.whl
下载地址:https://download.pytorch.org/whl/cu110/torchvision-0.8.2%2Bcu110-cp38-cp38-win_amd64.whl


下好轮子后,进入whl文件所在目录,输入:

pip install torch-1.7.1+cu110-cp38-cp38-win_amd64.whl
pip install torchvision-0.8.2+cu110-cp38-cp38-win_amd64.whl


比如我下载到了 D:downloads,就进入这个目录,输入上面的命令即可安装。

5 配置完成


在当前环境中,我们输入python,再输入import torch,没有报错则安装成功,之后输入torch.cuda.is_available(),返回True则可以调用GPU,于是配置成功!

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/969186.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

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

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