在Featurize上部署飞桨执行深度学习任务

在Featurize上部署飞桨执行深度学习任务该文介绍了如何导入数据集 检查和安装 CUDA 以及解决 CUDA 内存分配错误的方法

大家好,欢迎来到IT知识分享网。

1.导入数据集 

在Featurize上部署飞桨执行深度学习任务

2.配置

检查CUDA的版本号,如果指令没有出结果,说明未安装CUDA

ls -l /usr/local | grep cuda

在Featurize上部署飞桨执行深度学习任务

 在下面的链接中找到对应的版本进行安装

开始使用_飞桨-源于产业实践的开源深度学习平台 (paddlepaddle.org.cn)

3.创建虚拟环境

使用原本的环境下载volumentations库时发生了冲突,遂搭建新的虚拟环境

conda create --prefix /cloud/yonu python=3.8 conda activate /cloud/yonu

注:代码中的”yonu”和”python=3.8″根据自己的情况进行修改

4.在 jupyter notebook 中使用虚拟环境

待更新…

5.安装CUDNN 

cuDNN Download | NVIDIA Developer

出现内存分配错误,输入下面的指令

export FLAGS_use_cuda_managed_memory=false

免责声明:本站所有文章内容,图片,视频等均是来源于用户投稿和互联网及文摘转载整编而成,不代表本站观点,不承担相关法律责任。其著作权各归其原作者或其出版社所有。如发现本站有涉嫌抄袭侵权/违法违规的内容,侵犯到您的权益,请在线联系站长,一经查实,本站将立刻删除。 本文来自网络,若有侵权,请联系删除,如若转载,请注明出处:https://haidsoft.com/149207.html

(0)
上一篇 2025-03-25 15:33
下一篇 2025-03-25 15:45

相关推荐

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注

关注微信