Ubuntu 16.04 LTS 安装PyTorch

Ubuntu 16.04 LTS 安装PyTorch1.安装2.测试3.其它问题4.参考 1.安装 1.打开终端 2.运行命令(由于我的CUDA版本为9.0,所以安装1.1.0版本的PyTorch, 0.3.0版本的torchvision) pip install torch==1.1.0 torchvision==0.3.0 --user 安装完成后分别显示: Successfully installed torch-1.1.0 torchvision-0.3.0 这里,需要提醒大家: 1.需要先安装CUDA,且CUDA和PyTorch以及PyTorch和torchvision之间的版本