【ubuntu】安装cuda

来源:互联网 发布:淘宝客搜索计入权重吗 编辑:程序博客网 时间:2024/06/06 11:40

参考:https://github.com/NVIDIA/DIGITS/blob/digits-5.0/docs/InstallCuda.md#driver

#查看已安装驱动

cat /proc/driver/nvidia/version

NVRM version: NVIDIA UNIX x86_64 Kernel Module  384.90 

#卸载nvidia驱动

sudo apt-get remove --purge nvidia-384*

(用于ubuntu图形化界面循环登录,可先卸载nvidia驱动)

1,下载cuda驱动

http://www.nvidia.com/Download/index.aspx

2,安装 .deb文件

dpkg -i cuda-repo-ubuntu1404-7-5-local_7.5-18_amd64.debapt-get update
3,自动安装cuda toolkit和驱动

apt-get install cuda

或者:分别安装toolkit和drivers

  1. apt-get install cuda-toolkit-7-5 - Installs only the toolkit and not the driver.
  2. apt-get install cuda-drivers - Installs only the driver and not the toolkit.
4,配置环境变量

vi ~/.bashrc

##########添加内容###########

# CUDA
export LOCAL_DIR=/usr/local
export CUDA_75_HOME=$LOCAL_DIR/cuda-7.5
export CUDA_80_HOME=$LOCAL_DIR/cuda-8.0
export CUDA_HOME=$CUDA_80_HOME
export PATH=$CUDA_HOME/bin:$PATH
export LD_LIBRARY_PATH=$CUDA_HOME/lib64:$CUDA_HOME/lib:$LD_LIBRARY_PATH


export CUDA_PATH=$CUDA_80_HOME

###########end########

source ~/.bashrc

5,重启

6,测试

nvcc -V

nvidia-smi

(使用.run安装cuda及nvidia驱动时不要选择装opengl)


原创粉丝点击