加速安装pytorch

文章目录

加速安装pytorch
- 1 创建环境
- 2 显卡版本查询
- 3 安装合适的版本
- 4 加快安装速度
- 5 检验是否安装成功
- 6 参考链接

1 创建环境

打开Anaconda Prompt终端

#查看现在有哪些环境
conda env list#创建新环境, 环境名称为：pytorch ，python版本为3.7
conda create -n pytorch python=3.7#切换到该环境
conda activate pytorch#如果想切换到其他环境eg:base，可以使用
conda activate base#查看具体有哪些命令,使用help
conda help
conda create --help

2 显卡版本查询

找到NVIDIA控制面板

NVIDIA控制面板有的在控制面板中可以找到，有的直接在桌面右下角可以看到 ;

更方便的直接在win10左下角的搜索框搜：NVIDIA；

点击帮助 — 系统信息 — 组件 — NVCUDA.DLL 可以看到CUDA的版本为10.1

3 安装合适的版本

点击pytorch官网，如下图

安装GPU版本

选择对应的CUDA版本10.1 + python语言 + 安装方式(可选 pip / conda /源码安装 ) + 电脑系统(linux / Mac /Windows) + pytorch版本

#conda 安装
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch#pip安装
pip install torch===1.4.0 torchvision===0.5.0 -f https://download.pytorch.org/whl/torch_stable.html

安装CPU版本

如果不想安装GPU版本，CUDA选项设置为None

#conda 安装
conda install pytorch torchvision cpuonly -c pytorch#pip安装
pip install torch==1.4.0+cpu torchvision==0.5.0+cpu -f https://download.pytorch.org/whl/torch_stable.html

4 加快安装速度

按照官网命令进行安装时，速度很慢。这里推荐用清华源替代默认conda源的方法，可以解决下载过慢的问题

添加清华源

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes

为了保险起见，建议同时添加第三方conda源：

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

如果使用清华源进行安装，那么必须去掉 conda 安装命令后面的 -c pytorch，即:

#使用默认源
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch#使用清华源
conda install pytorch torchvision cudatoolkit=10.1

5 检验是否安装成功

#进入python环境
pythonimport torch
print(torch.__version__)   #查看torch版本
print(torch.cuda.is_available) #查看GPU版本是否可用
exit()  #退出#查看conda 安装了哪些库
conda list

能看到：cudatoolkit 版本为 10.1 pytorch版本为1.4.0；

安装成功！

6 参考链接

pytorch 在win10下的安装

conda安装Pytorch下载过慢解决办法

pytorch官网

CUDA官网

清华conda源地址

加速安装pytorch相关推荐

（最新最全）windows使用anaconda安装pytorch进行深度学习并使用GPU加速
本篇文章记录下自己安装pytorch的过程,由于我装过3~4次了,所以还算是比较有经验了. 文章目录 1.检查电脑配置 2.Anaconda的准备工作-添加国内镜像源 3.下载 3.测试 1.检查电脑 ...
Jetson Nano安装pytorch 基于torch1.6和torchvision0.7
需要注意的是,博主使用的是win10主机,通过局域网连接的jetson nano, 其中jetson nano的预制CUDA版本为10.2 Jetpack 4.1.1 分别执行以下命令,即可查看自己的 ...
如何安装PyTorch
1 什么是PyTorch PyTorch是一个基于Python的科学计算包,其类似于Numpy但是能利用GPU加速,也是一个非常灵活和快速用于深度学习的研究平台. 2 安装PyTorch 检查自己的电 ...
TX2安装pytorch+TensorRT+yolov5实现实时检测
已完成的环境配置: TX2刷机后,完成了opencv4.5.1的编译:Ubuntu18.04安装opencv4.5.1+contrib 支持cuda加速(附带编译好的opencv4.5.1及缺失文件) ...
ubuntu 16.04下安装pytorch配置caffe2（cuda 9.0+cudnn 7.6）
caffe2已经合并进啦pytorch里面,所以我们可以通过安装pytorch完成对caffe2的配置,相较于安装caffe2源码的方式,操作上更为简单. 操作系统: Ubuntu 16.04.5 L ...
莫烦python pytorch_[莫烦 PyTorch 系列教程] 1.2 – 安装 PyTorch
支持的系统 PyTorch 暂时只支持 MacOS, Linux. 暂不支持 Windows! (可怜的 Windows 同学们.. 又被抛弃了). 不过说不定像 Tensorflow 一样, 因为 ...
win10 上安装 pytorch + cuda
PyTorch 学习: 第1天: win10 上安装 pytorch + cuda 2022-09-15 by Zhang https://pytorch.org/tutorials/ 关于 PyTo ...
conda安装Pytorch下载过慢解决办法(11月26日更新ubuntu下pytorch1.3安装方法)
目录添加清华源安装PyTorch 3月5日更新ubuntu下pytorch1.0.1安装方法(Ubuntu16.04+CUDA9.0+PyTorch1.0.1) 7月23日更新ubuntu下pyt ...
使用Anaconda安装pytorch和tesorflow以及相关环境配置
Anaconda拥有非常强大的python包和库管理功能,并且机器学习常用的pytorch和tensorflow两个库可以直接使用anaconda安装,这两个库的官网分别是:https://pytor ...

加速安装pytorch

加速安装pytorch

文章目录

1 创建环境

2 显卡版本查询

3 安装合适的版本

4 加快安装速度

5 检验是否安装成功

6 参考链接

加速安装pytorch相关推荐

最新文章

热门文章