一、环境说明
环境 | 版本 |
系统 | windows10 |
显卡 | RTX3060 |
CUDA | 11.1.1 |
cuDNN | 8.1.1 |
Python | 3.8.8 |
torch | 1.9.0+cu111 |
torchvision | 0.10.0+cu111 |
torchaudio | 0.9.0 |
cuda安装参考: tensorflow-gpu2.4.1安装配置详细步骤
算法模型下载地址:链接:https://pan.baidu.com/s/1uGNXd5g2dKjb4zgckxiLdA
提取码:vqv6
二、安装依赖模块