记录linux系统安装GPU版pytorch

记录linux系统安装GPU版pytorch

1、确认服务器是否有显卡

执行如下命令：

lspci | grep -i nvidia

1）如果没有任何输出信息则代表服务器中没有显卡
2）如果输出NVIDIA相关信息则代表服务器中有显卡

2、安装显卡驱动

1）在 NVIDIA官网依据自己的显卡型号下载对应的驱动，需要选择对应的CUDA版本，记住自己所选的CUDA版本，后续会使用到。

2）下载得到run文件，上传至服务器后执行以下命令安装驱动：

./xxxxxx.run  #xxxxxx.run为驱动的文件名，并且没有空格

直接Enter就好

3）验证安装是否成功
使用如下命令：

nvidia-smi

如输出如下图所示信息则代表安装成功

PS：图片来源于网络。

3、安装Anaconda3

下载Anaconda3上传至服务器，执行以下命令：

sh xxxxxx.run  #xxxxxx.run为驱动的文件名

同样直接Enter即可

安装完成后配置anaconda的环境变量，参考配置Anaconda3环境变量

PS：安装Anaconda3以后就无需再单独安装CUDA！！Anaconda3自带CUDA！！！

4、安装Pytorch

依据下载NVIDIA驱动时所选择的CUDA版本，在pytorch官网查找相对应的安装命令，例如我的CUDA版本是10.0，则所需安装命令如下：

pip install torch==1.2.0 torchvision==0.4.0

至此安装成功！