Ubuntu20.04手动安装cuda11.3
Ubuntu20.04手动安装cuda11.3
·
官方下载地址
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
欢迎来到由智源人工智能研究院发起的Triton中文社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。
更多推荐
已为社区贡献1条内容
所有评论(0)