关键词

Ubuntu下安装pytorch(GPU版)

我这里主要参考了:https://blog.csdn.net/yimingsilence/article/details/79631567

并根据自己在安装中遇到的情况做了一些改动。

 

先说明一下我的Ubuntu和GPU版本:

Ubuntu 16.04

GPU:GEFORCE GTX 1060

 

1. 查看显卡型号

使用命令:lspci | grep -i nvidia 查看电脑上的显卡,是否是nvidia版本。

2. 安装NVIDIA显卡驱动

具体做法参考之前博客的介绍。

3. 安装cuda 8

下载地址:https://developer.nvidia.com/compute/cuda/8.0/Prod2/local_installers/cuda_8.0.61_375.26_linux-run

下载下来之后是一个.sh文件,运行这个文件就开始安装了,后面会出现条款,全部同意即可。不过中间有一步,如果已经通过2自己安装过驱动了就不用安装了,如果没有就安装。

4. 设置cuda的环境变量

在~/.bashrc中的末尾添加:

export CUDA_HOME=/usr/local/cuda-8.0
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH
export PATH=/usr/local/cuda-8.0/bin:$PATH

然后source ~/.bashrc立即生效。

5. 检查和验证之前的操作是否正确

(1)使用命令:nvcc --version,查看nvcc版本;

(2)使用命令:nvidia-smi,查看驱动是否安装成功,这一步之前可能需要reboot一下使得刚装的驱动生效;

(3)运行一下官方的例子。进入到samples文件夹(在用户目录下或者/usr/local/cuda/samples下都有),进入1_Utilities/deviceQuery中,make 然后运行./deviceQuery,会输出一堆信息,如果最后是PASS,则说明之前都对。

6. 安装cudnn

官网下载地址:https://developer.nvidia.com/rdp/cudnn-download

我选择了cuDNN v5.1 Library for linux,下载之后使用命令:tar -xvf filename,解压即可。然后把lib和include拷贝到cuda目录:

sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64

然后修改拷过去的文件的访问权限:
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

然后命令行进入 cuda/lib64 路径下,运行以下命令:

sudo cp lib* /usr/local/cuda/lib64/ #复制动态链接库
cd /usr/local/cuda/lib64/sudo rm -rf libcudnn.so libcudnn.so.5 #删除原有动态文件
sudo ln -s libcudnn.so.5.1.10 libcudnn.so.5 #生成软衔接
sudo ln -s libcudnn.so.5 libcudnn.so #生成软链接

最后更新链接库:sudo Idconfig

到这里cuda和cudnn就安装完成了,下面开始anaconda和pytorch的安装

7. anaconda安装

下载地址:https://www.anaconda.com/download/

下载下来,是一个.sh文件,然后运行这个命令即可(运行命令./sh)。

设置虚拟环境:conda create -n env_name python=3.6.5以创建一个虚拟环境env_name。

使用命令:conda env_name,来激活虚拟环境

8. 安装pytorch

下载地址:https://pytorch.org/get-started/previous-versions/

我选择下载的是:https://download.pytorch.org/whl/cu80/torch-0.4.1-cp36-cp36m-linux_x86_64.whl

然后使用命令安装:pip install torch-0.4.1-cp36-cp36m-linux_x86_64.whl

再安装torchvision:pip install torchvision

9. 测试安装是否成功

import torch
import torchvision
如果没报错就说明安装成功了。

本文链接:http://task.lmcjl.com/news/5651.html

展开阅读全文