以titan rtx为例 在ubuntu server上部署cuda+cudnn+anaconda

之前经常自己搭anaconda,在linux和windows上搭过。上个月给实验室服务器搭环境供大家使用,想着总结一下吧,省得以后再到处找。。。

 

先装个ubuntu server 然后改网络配置

ubuntu server18把网络配置变了,网上也能查到  静态配置在

 /etc/netplan/50-cloud-init.yaml

配置好网络了,我们开始装显卡驱动

显卡驱动要去nvidia官网找最好

先禁用Nouveau驱动

sudo apt-get update

sudo gedit /etc/modprobe.d/blacklist.conf

在文件最下面添加:

blacklist nouveau

执行:

sudo update-initramfs -u

然后reboot,重启之后看看是否禁用成功

lsmod |grep nouv*  

去nvidia官网找到linux版本的驱动,下载下来的是.sh的文件,执行即可

安装好后用指令验证

nvidia-smi

去官网下载cuda对应版本,也是.sh文件,执行即可

安装的时候把driver选项去掉,因为我们已经装过驱动了

安装好后开始改环境变量:

~/.bashrc里添加

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

然后保存,执行使其生效

source ~/.bashrc

cudnn需要找到对应cuda版本的去下载

解压后,进行一些复制操作(本质上跟windows差不多,替换一些cuda的文件):

#在解压出来的路径中寻找对应文件夹
sudo cp /include/cudnn.h  /usr/local/cuda/include/
sudo cp lib*  /usr/local/cuda/lib64/
 
#设置权限
sudo chmod a+r /usr/local/cuda/include/cudnn.h 
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
 
#更新软链接
cd /usr/local/cuda/lib64/ 
sudo rm -rf libcudnn.so libcudnn.so.7   #注意版本号,可在cudnn的lib64文件夹中查看   
sudo ln -s libcudnn.so.7.6.2 libcudnn.so.7  #生成软衔接,也要注意版本号
sudo ln -s libcudnn.so.7 libcudnn.so
sudo ldconfig -v

nvcc -V测试即可

 

安装anaconda,在anaconda中新建环境,安装tensorflow,pytorch等

anaconda安装完也要修改环境变量

在base中安装了jupyter,在每个环境中安装ipykernel,然后:

/xxxx/envs/xxx/bin/python -m ipykernel install --name xxx

即可将该环境添加至jupyter的kernel中

jupyter的远程配置需要生成自己的密码的hash值,然后修改配置文件里的一系列配置即可

jupyter里面还有latex编译生成,这个坑以后再添

在Windows系统下,针对RTX 5070显卡运行YOLOv4时,CUDAcuDNN的版本选择需综合考虑显卡驱动支持、PyTorch(或其他深度学习框架)兼容性以及YOLOv4本身的依赖要求。以下是具体的配置建议: ### CUDA 版本选择 RTX 5070属于NVIDIA新一代Ampere架构的消费级显卡,其对CUDA的支持通常从11.x开始。根据类似显卡如RTX 3060的经验,推荐使用 **CUDA 11.6 或 11.8**,这两个版本在稳定性和兼容性方面表现良好[^4]。此外,CUDA 11.6是PyTorch 1.11等主流深度学习框架所支持的版本之一,能够满足YOLOv4训练与推理的需求。 ### cuDNN 版本选择 对于CUDA 11.6或11.8,推荐使用 **cuDNN 8.4 或更高版本**,因为该版本提供了对现代神经网络操作的良好支持,并且与大多数深度学习框架兼容[^2]。确保cuDNN版本与CUDA Toolkit版本匹配,避免因版本不一致导致的编译错误或运行时问题。 ### 显卡驱动安装 在安装CUDA之前,必须先安装合适的NVIDIA显卡驱动程序。可以通过[NVIDIA官网](https://www.nvidia.cn/drivers/lookup/)下载适用于RTX 5070的最新驱动。建议使用**Studio Driver**或**Game Ready Driver**,具体取决于应用场景(如科研计算或游戏开发)。 ### 环境变量设置 安装完CUDA后,需要将CUDA相关的路径添加到系统的环境变量中,如`C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6\bin`和`C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6\libnvvp`,以确保系统可以正确识别CUDA工具链和库文件[^3]。 ### 验证CUDA安装 为了验证CUDA是否安装成功,可以在命令行中进入CUDA安装目录下的`\extras\demo_suite`文件夹,然后分别运行 `bandwidthTest.exe` 和 `deviceQuery.exe`。如果输出结果中显示 `Result = PASS`,则表示CUDA已正确安装并配置[^3]。 ### YOLOv4 编译与部署 YOLOv4 的官方实现主要基于Darknet框架,它支持CUDA加速。为确保YOLOv4能够在RTX 5070上顺利运行,建议从Alexey Bochkovskiy的GitHub仓库克隆Darknet源码,并按照文档说明进行编译。编译过程中应启用GPU支持,即设置`GPU=1`、`CUDNN=1`以及`CUDNN_HALF=1`等选项,以便充分利用GPU资源提升性能[^3]。 ### 示:修改Makefile启用GPU支持 ```makefile GPU=1 CUDNN=1 CUDNN_HALF=1 ``` ### 安装PyTorch(可选) 如果计划使用基于PyTorch的YOLOv4实现(如某些第三方变种),请确保安装的PyTorch版本与当前CUDA版本兼容。如,若选择了CUDA 11.6,则可尝试安装PyTorch 1.11或1.12,这些版本均宣称对CUDA 11.6有较好的支持[^4]。 ---
评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值