在ubuntu系统配置cuda
时间: 2024-10-15 16:22:19 浏览: 46
在Ubuntu系统上配置CUDA通常需要以下步骤:
1. **安装依赖**:
- 首先,确保你的Ubuntu版本是最新的,并安装一些必要的包,如`gcc`, `make`, 和 `apt-get update`。
2. **下载CUDA**:
访问NVIDIA官网下载适用于你的Ubuntu版本和显卡型号的CUDA Toolkit。下载并解压到一个目录下,例如 `/usr/local/cuda`.
3. **添加安装路径到环境变量**:
打开终端,编辑`~/.bashrc` 或者创建一个新的文件(如`.bash_profile`),然后添加类似下面的内容:
```
export CUDA_HOME=/usr/local/cuda
export PATH=$PATH:$CUDA_HOME/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$CUDA_HOME/lib64
```
4. **更新环境变量**:
- 保存文件并关闭,然后运行 `source ~/.bashrc` (如果是.bash_profile,可以使用`source ~/.bash_profile`)来使更改生效。
5. **安装CUDA Toolkit**:
使用`sudo` 权限安装CUDA,命令类似于:
```bash
sudo dpkg -i cuda-repo-<version>-main.deb
sudo apt-get install cuda-cudart-dev
```
将 `<version>` 替换为你下载的CUDA版本号。
6. **确认安装**:
查看CUDA版本和设备列表,运行:
```bash
nvcc --version
nvidia-smi
```
7. **驱动程序安装** (如果需要):
如果你的显卡驱动不是由CUDA自动安装的,也可以通过`nvidia-driver`包管理工具进行安装。
8. **设置CUDA Toolkit SDK路径**:
对于某些CUDA项目,你可能还需要设置CUDA Toolkit的头文件搜索路径,这通常是在`CMakeLists.txt` 中完成。
9. **测试安装**:
创建一个简单的CUDA C++或Python项目,检查能否成功编译和运行以验证是否正确配置了CUDA。
阅读全文