Ubuntu20.04+CUDA12.5+TensorRT10.2.0

CUDA:

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/12.5.1/local_installers/cuda-repo-ubuntu2004-12-5-local_12.5.1-555.42.06-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu2004-12-5-local_12.5.1-555.42.06-1_amd64.deb
sudo cp /var/cuda-repo-ubuntu2004-12-5-local/cuda-*-keyring.gpg /usr/share/keyrings/
sudo apt-get update
sudo apt-get -y install cuda-toolkit-12-5
sudo apt-get install -y cuda-drivers

TensorRT:

wget https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/10.2.0/local_repo/nv-tensorrt-local-repo-ubuntu2004-10.2.0-cuda-12.5_1.0-1_amd64.deb
os="ubuntu2004"
tag="10.2.0-cuda-12.5"
sudo dpkg -i nv-tensorrt-local-repo-${os}-${tag}_1.0-1_amd64.deb
sudo cp /var/nv-tensorrt-local-repo-${os}-${tag}/*-keyring.gpg /usr/share/keyrings/
sudo apt-get update
sudo apt-get install tensorrt

其它

dpkg -l | grep tensorrt
sudo apt-get remove --purge tensorrt

#调整nvcc版本
export PATH=/usr/local/cuda-12.5/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-12.5/lib64:$LD_LIBRARY_PATH

sudo apt-get --purge remove "cuda*"
dpkg -l | grep cuda
sudo dpkg --purge libcudart10.1:amd64
sudo dpkg --purge nvidia-cuda-dev
sudo dpkg --purge nvidia-cuda-doc
sudo dpkg --purge nvidia-cuda-gdb
sudo dpkg --purge nvidia-cuda-toolkit
sudo apt-get autoremove
sudo apt-get clean

### 如何在 Ubuntu 20.04 上安装 CUDA 12.5 #### 准备工作 确保系统已更新至最新状态并重启计算机。这一步骤可以防止因软件包过期而导致的潜在冲突。 ```bash sudo apt-get update && sudo apt-get upgrade -y ``` #### 安装依赖项 安装必要的依赖库,这些工具对于后续操作至关重要: ```bash sudo apt-get install build-essential dkms linux-headers-$(uname -r) -y ``` #### 添加 NVIDIA 软件源 为了获取最新的驱动程序和支持文件,需配置系统的 APT 源列表来包含来自 NVIDIA 的官方仓库地址[^1]。 ```bash distribution=$(. /etc/os-release;echo $ID$VERSION_ID) wget https://developer.download.nvidia.com/compute/cuda/repos/$distribution/x86_64/cuda-keyring_1.0-1_all.deb sudo dpkg -i cuda-keyring_1.0-1_all.deb sudo apt-get update ``` #### 下载 CUDA 12.5 安装器 访问[NVIDIA 官方网站](https://developer.nvidia.com/cuda-downloads),选择适合的操作系统版本(即 Ubuntu 20.04 LTS),然后找到对应于目标架构 x86_64 和所需 CUDA 版本 12.5 的本地 runfile 文件链接进行下载。 ```bash wget https://developer.download.nvidia.com/compute/cuda/12.5.0/local_installers/cuda_12.5.0_530.30.02_linux.run chmod +x cuda_12.5.0_530.30.02_linux.run ``` #### 执行安装过程 运行刚刚下载好的脚本来启动图形界面引导式的安装向导,在此期间可以选择是否要自动处理现有显卡驱动以及其他选项设置;如果担心会覆盖现有的 GPU 驱动,则建议手动指定仅安装 toolkit 组件而不改动其他部分[^2]。 ```bash sudo ./cuda_12.5.0_530.30.02_linux.run --toolkit ``` #### 设置环境变量 编辑 `~/.bashrc` 或者 `/etc/profile.d/cuda.sh` 来永久生效 PATH 及 LD_LIBRARY_PATH 中关于新加入路径的信息,以便命令行能够识别 nvcc 编译器及其他相关工具链的位置。 ```bash echo 'export PATH=/usr/local/cuda-12.5/bin${PATH:+:${PATH}}' >> ~/.bashrc echo 'export LD_LIBRARY_PATH=/usr/local/cuda-12.5/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}' >> ~/.bashrc source ~/.bashrc ``` #### 测试安装成果 通过编译一个小例子验证整个流程是否顺利完成,比如下面这个简单的 Hello World 程序就可以作为测试对象之一。 ```cpp // hello_world.cu #include <stdio.h> __global__ void say_hello() { printf("Hello, world!\n"); } int main(void){ say_hello<<<1,1>>>(); cudaDeviceSynchronize(); } ``` 保存上述代码到名为 `hello_world.cu` 的文件里之后执行如下指令完成构建与运行: ```bash nvcc -o hello_world hello_world.cu ./hello_world ``` 如果一切正常的话应该可以在终端看到输出 “Hello, world!” 字样,这就意味着 CUDA 已经成功部署到了当前环境中[^3]。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值