ubuntu 调用cuda
时间: 2023-07-13 22:34:27 浏览: 96
要在 Ubuntu 中使用 CUDA,您需要遵循以下步骤:
1. 安装 NVIDIA 显卡驱动程序:在 Ubuntu 中,您可以使用以下命令安装 NVIDIA 显卡驱动程序:
```
sudo apt-get install nvidia-XXX
```
其中,XXX 是您的 NVIDIA 显卡驱动程序版本号。
2. 安装 CUDA 工具包:您可以从 NVIDIA 的官方网站下载 CUDA 工具包。下载后,您可以使用以下命令安装 CUDA 工具包:
```
sudo dpkg -i cuda-repo-ubuntuXX.XX_XX.XX.XX-1_amd64.deb
sudo apt-get update
sudo apt-get install cuda
```
其中,XX.XX 是您的 Ubuntu 版本号,XX.XX.XX 是 CUDA 工具包版本号。
3. 配置环境变量:在安装完成后,您需要将 CUDA 的路径添加到您的环境变量中。您可以将以下内容添加到您的 ~/.bashrc 文件中:
```
export PATH=/usr/local/cuda-XX.X/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-XX.X/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
```
其中,XX.X 是您的 CUDA 版本号。
4. 测试安装:您可以使用以下命令测试您的 CUDA 安装是否成功:
```
nvcc -V
```
如果您看到了 CUDA 版本号,则说明您的 CUDA 安装成功。
阅读全文