官方下载地址

CUDA Toolkit 11.6 Update 2 Downloads | NVIDIA Developer

Ubuntu 11.3.0版本

1.为什么手动安装

以前在windows上写代码的时候,要安装显卡驱动,cuda toolkit和cDNN,到了Ubuntu上,只用安装驱动,因为pytorch上可以安装自带的cuda库,但是在跑别人的代码的时候,别人自己指定了要利用c++结合cuda来计算,不用pytorch,那么cuda库就用不了额,因为cuda库在pytorch中,只好手动安装,否则一直找不到CUDA_HOME这个环境变量。

1.按照官网的安装命令安装,但是不要安装cuda自带的显卡驱动程序,按空格取消

取消cuda自带的驱动

 

选择install 安装

2.添加环境变量

export PATH=/usr/local/cuda-11.3/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-11.3/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
export CUDA_HOME=/usr/local/cuda-11.3:/usr/local/cuda

3.查看安装测试结果

使用命令查看

>nvcc -V

卸载CUDA教程

卸载命令

# cuda10.1及以上的卸载
cd /usr/local/cuda-xx.x/bin/
sudo ./cuda-uninstaller
sudo rm -rf /usr/local/cuda-xx.x

Logo

欢迎来到由智源人工智能研究院发起的Triton中文社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。

更多推荐