一、工具安装

sudo apt-get install gcc
sudo apt-get install g++
sudo apt-get install make

二、禁用nouveau(必须的)
    sudo gedit /etc/modprobe.d/blacklist.conf
    添加 blacklist nouveau
    sudo update-initramfs -u
    sudo reboot
    可以通过  lsmod | grep nouveau   检查是否已禁用。

三、安装cuda(装nvdia显卡时,确定不在lightdm环境下,gdm3可以)
    1.如果在lightdm下,停止lightdm串口管理器,之后会进入黑窗口,建议停止之前拍照接下来的操作。
    sudo service lightdm stop (or:sudo systemctl stop  lightdm)
    2.清除旧版本的nvidia驱动和cuda:(推荐安装使用cuda***.run文件安装驱动和cuda;也可以单独安装驱动,在run文件安装过程去除驱动安装,但是得保证版本对应)

    通过run文件安装:

sudo /usr/local/cuda-10.2/bin/cuda-uninstaller
sudo /usr/bin/nvidia-uninstall

    通过apt安装:

sudo apt-get --purge remove "*cublas*" "cuda*"
sudo apt-get --purge remove "*nvidia*"

    3.下载cuda(推荐run文件)
    sudo sh cuda**.run
    4.查看显卡信息,验证驱动是否安装成功
    nvidia-smi
    5.sudo service lightdm start
    6.cd usr/local

        

          

    查看具体的路径,上面两个图圈住的名字不一样,对应自己的版本,更改下面export的内容。
    7.sudo gedit ~/.bashrc
    添加行(如果不加nsight-compute路径,使用nsys会出现没有此命令

export PATH=/usr/local/cuda-10.2/bin:/usr/local/cuda-10.2/nsight-compute-2019.5.0/${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64/${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

    8.source ~/.bashrc
    9.检查是否安装成功的几个方法:
        1.nvcc -V
        2.cat /proc/driver/nvidia/version
        3.cd /usr/local/cuda/samples/1_Utilities/deviceQuery
          sudo make
          ./deviceQuery
        4.nsys --version

四、安装cudnn
    可以官网下载deb的文件,直接双击安装,但是需要注册账号

五、安装pytorch
    进入官网下载即可。

    注:如果网络不好,建议使用pip安装,手动在浏览器下载pytorch的whl文件(700M,网速不好很容易中断),然后本地安装。
    手动安装whl文件的方法:
    1.安装依赖库 conda install numpy matplotlib py-opencv
    2.pip install /home/liu/package/pytorch/torch-1.1.0-cp37-cp37m-linux_x86_64.whl

六、检查是否安装成功

import torch
print(torch.__version__)
print(torch.cuda.is_available())

conda list

pip list 

参考:

官方linux安装教程:

pdf版:https://developer.download.nvidia.com/compute/cuda/7.5/Prod/docs/sidebar/CUDA_Installation_Guide_Linux.pdf

网页版:

https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html

Logo

技术共进,成长同行——讯飞AI开发者社区

更多推荐