直接 pip install pytorch 是 cpu 版本的，不能用 gpu，得用下面这句话安装。
pip3 install torch==1.4.0+cu100 torchvision==0.5.0+cu100 -f https://download.pytorch.org/whl/torch_stable.html
1.4.0 是版本，这个地方输入错了没关系，会告诉你可用的版本。
cu100 表示 cuda 版本 10.0
直接 pip install pytorch 是 cpu 版本的，不能用 gpu，得用下面这句话安装。
pip3 install torch==1.4.0+cu100 torchvision==0.5.0+cu100 -f https://download.pytorch.org/whl/torch_stable.html
1.4.0 是版本，这个地方输入错了没关系，会告诉你可用的版本。
cu100 表示 cuda 版本 10.0