HiDDeN 模型的环境复现细节

前言

🤟 找工作，来万码优才：👉 #小程序://万码优才/r6rqmzDaXpYkJZF

模型配置的具体说明：如何根据 CUDA 配置安装 PyTorch 和 torchvision（大模型 环境经验）

1. 配置

今天部署的大模型链接：https://github.com/ando-khachatryan/HiDDeN

项目只需要python3.6以上，torch为1.0以上即可

创建虚拟环境：conda create -n name python=3.7
根据自动的CUDA安装对应的版本，主要是torch以及torchvision的差异

自身CUDA是13版本，对应的版本如下：