以下是详细的“Ubuntu下安装CUDA10.0以及问题”的完整攻略:
1. 准备工作
在安装CUDA 10.0之前,需要具备以下条件:
- Ubuntu操作系统,推荐使用18.04或更高版本
- 安装完整的NVIDIA显卡驱动程序
- 支持CUDA的NVIDIA显卡
如果您还没有安装NVIDIA显卡驱动程序,请先参考NVIDIA官方文档进行安装。
2. 下载CUDA安装包
访问 CUDA Toolkit Archive 页面,找到CUDA 10.0版本,并找到适合您操作系统的安装包下载地址。
我们可以使用命令行下载CUDA 10.0安装包:
$ wget https://developer.download.nvidia.com/compute/cuda/10.0/secure/Prod/local_installers/cuda-repo-ubuntu1804-10-0-local-10.0.130-410.48_1.0-1_amd64.deb
3. 安装CUDA
3.1 安装依赖
在安装CUDA安装包之前,需要安装必要的依赖文件:
$ sudo apt-get update
$ sudo apt-get install build-essential dkms freeglut3-dev libxi-dev libxmu-dev #安装必要的依赖
3.2 安装CUDA
运行以下命令安装CUDA 10.0:
$ sudo dpkg -i cuda-repo-ubuntu1804-10-0-local-10.0.130-410.48_1.0-1_amd64.deb
$ sudo apt-key add /var/cuda-repo-<version>/7fa2af80.pub
$ sudo apt-get update
$ sudo apt-get install cuda
4. 配置环境变量
完成CUDA安装后,需要设置CUDA的环境变量,以便系统能够找到CUDA的库文件和工具。
在终端输入以下命令,打开.bashrc文件:
$ gedit ~/.bashrc
在.bashrc文件的最后添加以下代码:
export PATH=/usr/local/cuda-10.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64:$LD_LIBRARY_PATH
保存并关闭文件,运行以下命令使配置生效:
$ source ~/.bashrc
5. 测试CUDA
在终端中输入以下命令测试CUDA是否安装成功:
$ nvcc -V
如果输出类似以下内容,说明CUDA已经正常安装:
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2018 NVIDIA Corporation
Built on Tue_Jun_12_23:07:04_CDT_2018
Cuda compilation tools, release 10.0, V10.0.130
在终端中输入以下命令进行CUDA样例测试:
$ cd /usr/local/cuda/samples
$ sudo make -j4 # 编译 CUDA 示例程序
$ cd /usr/local/cuda/samples/bin/x86_64/linux/release
$ ./deviceQuery # 运行 CUDA 示例程序
$ ./bandwidthTest
运行样例程序如果输出正确的结果,则说明CUDA安装成功。
6. 解决问题
在安装CUDA过程中,可能会遇到以下问题及其解决方法。
6.1 安装CUDA时无法升级nvidia-driver
解决方法:使用系统自带的nvidia驱动,而不是手动安装。
6.2 安装CUDA后Ubuntu系统变得不稳定
解决方法:卸载CUDA并重启电脑,使用新版本的CUDA或者安装其他可选版本。
示例
示例一:安装CUDA 10.0并测试
用户在Ubuntu 18.04系统上,使用GTX 1070显卡,已经安装完整的NVIDIA显卡驱动程序,准备安装CUDA 10.0并测试。
用户打开终端,按照上述攻略中的步骤,完成安装CUDA 10.0及相关依赖。安装完成后,用户输入以下命令测试CUDA:
$ nvcc -V
终端输出如下内容:
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2018 NVIDIA Corporation
Built on Tue_Jun_12_23:07:04_CDT_2018
Cuda compilation tools, release 10.0, V10.0.130
$ cd /usr/local/cuda/samples
$ sudo make -j4
$ cd /usr/local/cuda/samples/bin/x86_64/linux/release
$ ./deviceQuery
终端输出类似以下内容:
CUDA Device Query (Runtime API) version (CUDART static linking)
Detected 1 CUDA Capable device(s)
Device 0: "GeForce GTX 1070"
CUDA Driver Version / Runtime Version 10.0 / 10.0
CUDA Capability Major/Minor version number: 6.1
Total amount of global memory: 8119 MBytes (8518623232 bytes)
(...)
以上结果表明,用户在Ubuntu 18.04系统上成功安装了CUDA 10.0,并通过样例程序测试成功。
示例二:升级CUDA版本
用户在Ubuntu 20.04系统上,使用RTX 3080显卡,已经安装完整的NVIDIA显卡驱动程序,但需要安装新版本的CUDA,因为新版本支持RTX 3080显卡。
用户打开终端,按照上述攻略中的步骤,卸载原有的CUDA,并下载新版本CUDA安装包。安装完成后,用户输入以下命令测试CUDA:
$ nvcc -V
终端输出类似以下内容:
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2021 NVIDIA Corporation
Built on Wed_Jun__2_19:15:15_PDT_2021
Cuda compilation tools, release 11.4, V11.4.48
Build cuda_11.4.r11.4/compiler.30082316_0
$ cd /usr/local/cuda/samples
$ sudo make -j4
$ cd /usr/local/cuda/samples/bin/x86_64/linux/release
$ ./deviceQuery
终端输出类似以下内容:
CUDA Device Query (Runtime API) version (CUDART static linking)
Detected 1 CUDA Capable device(s)
Device 0: "GeForce RTX 3080"
CUDA Driver Version / Runtime Version 11.4 / 11.4
CUDA Capability Major/Minor version number: 8.6
Total amount of global memory: 10018 MBytes (10473719808 bytes)
(...)
以上结果表明,用户在Ubuntu 20.04系统成功升级到新版本CUDA,并通过样例程序测试成功。
本站文章如无特殊说明,均为本站原创,如若转载,请注明出处:Ubuntu下安装CUDA10.0以及问题 - Python技术站