pytorch 1.1+win10 +python3.67+cuda10.0 安装
点击进入官网 https://pytorch.org/
pip安装
Python3.68+pip安装cpu版本
pip3 install https://download.pytorch.org/whl/cpu/torch-1.1.0-cp36-cp36m-win_amd64.whl
pip3 install torchvision
Python3.68+pip安装gpu版本
目前gpu版本支持cuda8.0,cuda9.0和cuda10.0,请选择对应的版本下载安装,不要同时执行下面三个命令!喜欢conda的可以选择conda安装
pip3 install https://download.pytorch.org/whl/cu100/torch-1.1.0-cp36-cp36m-win_amd64.whl
pip3 install torchvision
检测是否成功!
import torch
print(torch.cuda.is_available())
如果返回True
则可以使用,经测试可以正常使用
不得不说,比tensorflow 的安装方便了100倍!
还有问题的,可以加群264191384 注明(pytorch)
-