TI-ONE平台安装cuda指引:https://cloud.tencent.com/developer/article/1845781
先切到root(sudo su root) 再安装,安装失败看看安装日志。该指引只测试过安装cuda10.1,其他版本没测试过。但TI-ONE不支持安装Nvidia 驱动,因此大家需要根据现有驱动版本 来看具体可以安装哪些cuda版本。
① 自定义环境可执行conda install cudnn cudatoolkit=10.1,版本可按照所需库对应关系设定,安装完可从对应conda环境的lib中看到库文件
② 使用系统默认环境(非conda环境配置)时,/usr/local/下是没有cuda库文件的,若需映射到/usr/local/cuda 路径下,则按照cuda安装指引完成cuda安装,在nvidia官网下载对应cudnn版本解压, 拷贝lib64目录和include/cudnn.h文件到/usr/local/cuda/lib64/和/usr/local/cuda/include/,然后配置环境并更新即可。
自定义conda环境的cuda需正确安装,可使用conda install cudnn cudatoolkit=10.1命令安装后,用pip install onnxruntime-gpu==1.2命令安装1.2版本,经验证1.2版本可用
onnxruntime-gpu版本与cuda,cudnn的版本关系详见:https://www.onnxruntime.ai/docs/reference/execution-providers/CUDA-ExecutionProvider.html#requirements
需根据cuda版本(系统默认为cuda 10.1版本)来选择相应的torch安装版本,具体安装命令可查询 https://pytorch.org/get-started/previous-versions/
附1.7.1安装命令:
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.1 -c pytorch
备注:如是GPU调用出错问题,请附报错log。
系统默认cuda 10.1版本对应的tf版本有2.1(python版本2.7,3.5~3.7),2.2(python版本3.5~3.8),2.3(python版本3.5~3.8),默认tf虚拟环境安装了cuda 10.0 cudnn7.6 tf2.0,所以只有2.0能用gpu,自定义虚拟环境时需要在conda环境下重新安装对应的cuda cudnn tf版本
具体版本关系可查看tf官网https://tensorflow.google.cn/install/source
① conda环境的cuda库文件存储于conda环境所在目录的lib目录中,可以自行拷贝过去并配置环境即可
② 若需在/usr/local/下的cuda文件,则需要正确安装cuda及对应的cudnn,才可以正常使用GPU
不支持
不支持
不支持
不支持debug
计算量少的情况下GPU较CPU耗时长
传送门
信息化2.0时代提出开展智慧教育创新发展行动。2019年2月,中共中央、国务院印发...
从 10.0.0 版开始,异步迭代器就出现在 Node 中了,在本文中,我们将讨论异步迭...
Docker生成新镜像版本的两种方式 There are two ways Docker can generate new m...
前提条件 请您在购买前确保已完成注册和充值。详细操作请参见 如何注册公有云管...
【51CTO.com快译】 数据可视化工具不断发展,提供更强大的功能,同时改善可访问...
摘要 元旦期间 订单业务线 告知 推送系统 无法正常收发消息,作为推送系统维护者...
本文整理自直播《Hologres 数据导入/导出实践-王华峰(继儒)》 视频链接: https:/...
在Python语言中有如下3种方法: 成员方法 类方法(classmethod) 静态方法(staticm...
2021年3月24日,主题为《数据的世界,世界的数据》的星环科技2021春季新品发布会...
建站 什么 虚拟主机 够用?这要看搭建的是什么类型的网站。比如个人博客类型的网...