# 一些参考链接
- windows10 下安装 GPU 版 pytorch 简明教程
- GPU 版本的 PyTorch 安装与环境配置
# 安装教程
进入官网 https://pytorch.org/
找到安装文档
发现没有自己想要的版本(有的话,直接运行下面的命令就可以了),就自己去下载页面找到自己的版本
我的配置是 python3.6.5 + cuda10 + windows10
下载到本地(速度慢的就开 vpn)
本地安装
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" |
可以看到本地安装速度是非常快的
- 测试
import torch | |
torch.cuda.is_available() | |
# 打印 true 安装完美成功 |