直接使用 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。