RTX3080安装DEEPLABCUT使用GPU(win11)

步骤一:安装CUDA和CUDNN

参考教程:CUDA和CUDNN的安装和配置检查

注意:因为我的主机是RTX3080 只支持CUDA11.0及以上 这里就安装了11.0的CUDA
而且CUDA版本要跟CUDNN版本对应起来 我选的CUDNN是8.0的 具体安装和环境配置检查看上面的参考教程
如果是之前已经有别的版本的CUDA不用删除 可以多CUDA存在 具体参考教程----> 在win下安装多个版本的CUDA
CUDA官网链接
CUDNN官网链接

步骤二:安装DEEPLABCUT

参考官网:deeplabcut安装教程官网链接
1)把官网给的DEEPLABCUT.yaml文件下载下来 可以点开yaml这个文件看看 里面写了下载什么包
2)打开anaconda prompt进入DEEPLABCUT.yaml文件的下载目录 也可以直接把这个文件移到base环境初始目录就不用切换了
3)输入 conda env create -f DEEPLABCUT.yaml 就可以把deeplabcut环境需要的包安装好了 环境名字是DEEPLABCUT
4)激活环境 activate DEEPLABCUT

注意:官网给的安装文件里面默认安装最高版本的tensorflow 我安装下来里面python是3.8
可以用 conda list tensorflow 以及 conda list python 检查一下这个官网文件下下来的 tensorflow 和 python 都是什么版本

步骤三:安装torch

官网给的yaml文件里面没有torch 后面会报错提示安装torch所以安装一下 不会影响其他包的正常使用
小注意:pip安装包的时候记得把翻墙关了

5)  pip安装torch包 指令:                    pip install torch

没有配置换源的 手动加个豆瓣源快一点 -i https://pypi.douban.com/simple 加在包后面

步骤四:安装tensorflow-gpu

官网给的yaml文件里面下下来的只有tensorflow(只用cpu训练) 没有tensorflow-gpu(这个是用gpu来训练)
注意:tensorflow版本和tensorflow-gpu要一致 同时tensorflow-gpu的版本要跟自己的CUDA对应 另外注意一下python版本
参考:tensorflow和CUDA以及python对应关系官网链接
我的CUDA是11.0对应的tensorflow-gpu是2.4.0 所以yaml文件下载下来的tensorflow版本过高 需要自己手动安装需要的版本 同时tensorflow的版本要跟tensorflow-gpu的版本一致 不然会有版本冲突的报错 安装新的就可以把之前旧的自动uninstall

 6)pip安装tensorflow:                 pip install tensorflow==2.4.0 

7) pip安装tensorflow-gpu: 			  pip install tensorflow-gpu==2.4.0

步骤五:使用deeplabcut两种方式

方式一:使用deeplabcut GUI训练评估创建视频

C:\Users\xxx>  ipython
	    In[1]:  import deeplabcut
	    In[2]:  deeplabcut.launch_dlc()

方式二:使用终端 参考指令链接----> 指令在该网页的图片中
指令中参数的具体解释参考链接
官方视频演示deeplabcut
注意:在创建训练集的时候要选择需要的预训练模型 每个模型第一次使用需要下载 这里要翻墙 不然下不下来连接出错

  • 2
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 11
    评论
RTX 3050是NVIDIA推出的一款显卡,它基于Ampere架构,采用了功能强大的CUDA核心和Tensor核心,拥有较高的性能和能效比。在Win11操作系统下,RTX 3050能够充分发挥其显卡性能,提供更流畅的图形渲染和计算能力。 PyTorch是一种流行的用于深度学习的框架,它提供了丰富的功能和灵活的接口,方便用户进行模型设计、训练和部署。在使用PyTorch进行深度学习任务时,可以利用RTX 3050强大的GPU加速功能,加快模型的训练和推理速度,提高效率。 使用RTX 3050配合Win11操作系统和PyTorch框架进行深度学习任务,可以享受到以下优势: 1. 强大的计算性能:RTX 3050搭载强大的GPU,能够进行大规模的计算任务,加速深度学习模型的训练和推理过程,并通过提高每秒浮点运算次数(FLOPS)的速度来提高整体性能。 2. 高效的图形渲染:RTX 3050配备了优秀的图形渲染能力,能够提供更细致、逼真的图像效果,为计算机图形学和计算机视觉等任务提供更好的支持。 3. 易于开发和调试:PyTorch作为一种易于使用且社区活跃的深度学习框架,可以与RTX 3050很好地兼容。通过PyTorch的高级API和调试工具,可以快速搭建、训练和优化模型,实现个性化的深度学习任务。 总而言之,RTX 3050与Win11操作系统以及PyTorch框架的结合,为我们提供了一种强大的深度学习解决方案。无论是进行复杂的模型训练,还是进行高质量的图像渲染,它们的结合都能为用户带来更好的性能和使用体验。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 11
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值