pytorch官網(wǎng)地址
系統(tǒng)配置
- win10(1809)
- python3.6
pytorch版本
- stable 1.0
cuda版本
- cuda 10
安裝命令
因為沒有安裝conda悠垛,所以采用pip安裝
pip3 install https://download.pytorch.org/whl/cu100/torch-1.0.1-cp36-cp36m-win_amd64.whl
pip3 install torchvision
測試
import torch
torch.tensor([1,2])
安裝完成