目录
1.官网方式安装
2.利用国内镜像安装
如果你需要安装Pytorch环境,不要在使用官网推荐的方式,直接复制链接安装,使用清华镜像下载安装包,速度真的杠杠的。
1.官网方式安装
1.官网链接安装方式,直接百度Pytorch,进入官方网站,点击Get started 跳转指定位置
2.获取红线圈住的安装指令,打开doc窗口进行安装,将会龟速安装,国外网站拉取的是真的慢
2.利用国内镜像安装
本例演示使用conda方式安装pytorch,首先需要安装Anaconda,进入Anaconda官网https://www.anaconda/可以直接下载安装,此处略过。
1.首先打开Anaconda Prompt 创建pytorch环境
我的python为3.7版本
conda create -n pytorch python=3.7 指定自己的python版本即可。
2.更换镜像通道,将国外镜像换成清华镜像
输入命令:
conda config --add channels https://mirrors.tuna.tsinghua.edu/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu/anaconda/pkgs/main/
conda config --set show_channel_urls yes
conda config --add channels https://mirrors.tuna.tsinghua.edu/anaconda/cloud/pytorch/
3.最后按照你的cuda对应版本进行安装pytorch,我的电脑最高支持cuda9.2版本,所以我使用的cuda9.0版本,你可以自己查看自己电脑所支持的cuda最高版本
在桌面右击,选中NVIDIA 控制面板,进入控制面板
进去后点击底部系统信息
查看自己电脑所支持的最高版本,我的电脑比较菜,所以最高就支持到CUDA9.2版本
接下来进行安装所需要的pytorch环境
pytorch torchvision 也可以不指定版本号,使用默认版本
但是cudatoolkit指定自己适合的版本,否则默认安装最新版本,老显卡可能不支持
conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=9.0 -c python
4.最后可以进入python环境验证是否安装成功
- cmd进入doc,输入python,进入环境
- 导包 import torch
- 验证 print(torch.cuda.is_available())
- 结果返回True,成功
更多推荐
Win10系统利用国内镜像安装Pytorch--相比国外镜像不知道快多少倍
发布评论