Pytorch-Pytorch安装.md

Pytorch-Pytorch安装

一些参考链接

• windows10下安装GPU版pytorch简明教程
• GPU版本的PyTorch安装与环境配置

安装教程

1. 进入官网 https://pytorch.org/

2. 找到安装文档

3. 发现没有自己想要的版本（有的话，直接运行下面的命令就可以了），就自己去下载页面找到自己的版本

4. 我的配置是 python3.6.5 + cuda10 + windows10

5. 下载到本地（速度慢的就开vpn）

6. 本地安装

pip3 install "C:\Users\Administrator\Desktop\torch-1.2.0-cp36-cp36m-win_amd64.whl"
pip3 install "C:\Users\Administrator\Desktop\torchvision-0.4.0-cp36-cp36m-win_amd64.whl"

可以看到本地安装速度是非常快的

7. 测试

import torch
torch.cuda.is_available()
# 打印 true 安装完美成功