pytorch在pycharm中的安装

在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