在cmd中键入https://pytorch.org/get-started/locally/中选的命令行,
例安装cuad10.0,python3.7:
pip install https://download.pytorch.org/whl/cu100/torch-1.0.1-cp37-cp37m-win_amd64.whl --default-timeout=1000(后面加--default-timeout=1000可以防止read time错误)
pip3 install https://download.pytorch.org/whl/cu100/torchvision-0.3.0-cp37-cp37m-win_amd64.whl
在pycharm中输入
import torch
import torchvision
print(torch.cuda.is_available())
输出Ture证明可以使用gpu处理。
卸载pytorch
pip uninstall torch