Ubuntu下安装cuda和cudnn
时间: 2023-08-19 20:06:22 浏览: 121
ubuntu cuda cudnn
要在Ubuntu上安装CUDA和CuDNN,您可以按照以下步骤进行操作:
1. 安装CUDA:
a. 访问NVIDIA开发者网站(https://developer.nvidia.com/cuda-downloads)并下载适用于您的Ubuntu版本的CUDA安装包。
b. 打开终端,并切换到您下载的CUDA安装包所在的目录。
c. 运行以下命令解压安装包:
```
sudo sh cuda_*.run
```
d. 按照安装程序的指示进行安装。在安装过程中,您可以选择是否安装NVIDIA驱动程序和CUDA Toolkit。
2. 配置CUDA环境变量:
a. 打开终端,并打开`~/.bashrc`文件:
```
gedit ~/.bashrc
```
b. 在文件末尾添加以下行:
```
export PATH=/usr/local/cuda-<version>/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-<version>/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
```
注意将`<version>`替换为您所安装的CUDA版本号。
c. 保存并关闭文件。
d. 运行以下命令使更改生效:
```
source ~/.bashrc
```
3. 安装CuDNN:
a. 访问NVIDIA开发者网站(https://developer.nvidia.com/rdp/cudnn-archive)并下载与您所安装的CUDA版本兼容的CuDNN库文件。
b. 解压下载的CuDNN库文件。
c. 打开终端,并切换到解压后的CuDNN目录。
d. 运行以下命令将CuDNN库文件复制到CUDA安装目录:
```
sudo cp cuda/include/cudnn.h /usr/local/cuda-<version>/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-<version>/lib64
```
注意将`<version>`替换为您所安装的CUDA版本号。
现在,您应该已成功安装和配置了CUDA和CuDNN。
阅读全文