ubuntu18.04安装GPU PyTorch

转载自这篇文章

安装GPU版本的PyTorch

这里选择用pip进行安装，首先需要安装pip：执行命令sudo apt intall python-pip3

（该步骤可以跳过） 现在建议配置pip虚拟环境，为此我们需要配置virtualenv（有关virtualenv的详细介绍可以看看廖雪峰Python3教程和官方documentation）：

1. 安装virtualenv：执行命令pip3 install virtualenv
2. 新建一个虚拟环境（这里取名为env-pytorch)：在终端中切换到想要新建env-pytorch的目录下（创建完虚拟环境后此目录下会多一个名为env-pytorch的文件夹，包含该虚拟环境相关的全部文件），执行命令virtualenv --no-site-packages venv -p <PYTHON_EXE>，其中<PYTHON_EXE>是Python的版本号，例如python3.6
   到官网选择对应版本之后会得到相应的安装命令，如下图：
   
   

可以通过下面的命令进行测试：

python					#进入python
import torch
import torchvision

#返回当前设备索引
torch.cuda.current_device()

#返回GPU的数量
torch.cuda.device_count()

#返回gpu名字，设备索引默认从0开始
torch.cuda.get_device_name(0)

#cuda是否可用
torch.cuda.is_available()