ubuntu中cuda12.1配置（之前存在11.1版本的cuda）(同时配置两个版本)

向右看齐 2024-02-18 10:57 36阅读 0赞

#### ubuntu中cuda12.1配置 ####

由于YOLOv8项目中Pytorch版本需要cuda12.1版本  
![在这里插入图片描述][65461aa89f034085b7249cd3e433efb1.png]  
在官网下载12.1版本的deb包  
![在这里插入图片描述][c743d7b5399d4c899b36cea1d9c2815c.png]  
[官网地址][Link 1]

sudo dpkg -i cuda-keyring_1.0-1_all.deb
    sudo apt-get update
    sudo apt-get -y install cuda

然后需要修改bashrc文件（隐藏文件）  
![在这里插入图片描述][3a681ad49a57420a82149e59be55e126.png]  
添加

export PATH=/usr/local/cuda-12.1/bin:$PATH
    export LD_LIBRARY_PATH=/usr/local/cuda-12.1/lib64:$LD_LIBRARY_PATH

![在这里插入图片描述][81b88e0d471241b99c14cf47eb13138d.png]

保存  
![在这里插入图片描述][ec73c06479ea4ae69299a9dd292e981e.png]  
[参考文章][Link 2]

[65461aa89f034085b7249cd3e433efb1.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/02/18/c4ccb73f38e14dc390b92c8d09d311f8.png
[c743d7b5399d4c899b36cea1d9c2815c.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/02/18/3a93ca49290d4f66afcd60492756fafc.png
[Link 1]: https://developer.nvidia.com/cuda-toolkit-archive
[3a681ad49a57420a82149e59be55e126.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/02/18/9479dd805db5492599da99b8b318674c.png
[81b88e0d471241b99c14cf47eb13138d.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/02/18/e850cba871ca4544bf9dfec89e3475f5.png
[ec73c06479ea4ae69299a9dd292e981e.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/02/18/e6c880455ecc4d8783ccdf66cc098100.png
[Link 2]: https://www.jianshu.com/p/eb5335708f2a