
Vitis-AI 3.0 GPU版 PyTorch及优化器Docker镜像
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
本项目提供基于PyTorch框架的Vitis-AI 3.0 GPU版本Docker镜像,内置多种神经网络模型和优化器,便于深度学习应用快速开发与部署。
本资源提供Vitis-AI 3.0版本的docker镜像.tar文件下载链接,适用于NVIDIA显卡硬件平台。该镜像内置了pytorch量化编译器和优化器。使用方法:通过运行`docker load`命令将镜像文件导入系统后(可以利用`docker tag`指令进行重命名),按照官方手册中的指导步骤操作即可。
生成此Docker镜像是基于NVIDIA提供的nvidia/cuda:11.3.1-cudnn8-runtime-ubuntu20.04基础镜像,仅修改了apt、python和conda的下载源为国内版本,其余部分未作改动。参考的相关文章说明了上述dockerfile的具体修改内容。
全部评论 (0)
还没有任何评论哟~


