【pytorch】优化器
栏目:公司动态 发布时间:2024-04-29
本资源为Vitis-AI3.0版本docker镜像的.tar文件的下载链接,适用于NVIDIA显卡硬件平台,内置pytorch量化编译器镜像以及pytorch优化器镜像使用方法:使用dockerload指令将镜像文件导入后(导入后可以使用dock
本资源为Vitis-AI3.0版本docker镜像的.tar文件的下载链接,适用于NVIDIA显卡硬件平台,内置pytorch量化编译器镜像以及pytorch优化器镜像 使用方法: 使用docker load指令将镜像文件导入后(导入后可以使用docker tag指令改名),再按照官方手册中的使用方法即可 docker镜像生成过程: 按照官方github提供的3.0.0.001版本源代码中的dockerfile进行docker创建 以NVIDIA提供的nvidia/cuda:11.3.1-cudnn8-runtime-ubuntu20.04镜像为基础 仅修改apt、python、conda为国内下载源,其他未作变动 dockerfile修改内容: 参考文章https://blog.csdn.net/qq_36745999/article/details/129920225

平台注册入口