服务器选型
GPU为NVIDIA A10, 24G显存。GPU与CPU间支持PCIE 4.0连接. 支持TensorRT, RTX光追渲染等多种业务. 购买在镜像市场购买含vws镜像或自行加载NVIDIA GRID license后可支持专业级远程桌面应用.
GPU 计算型 gn7i
ecs.gn7i-c16g1.4xlarge
16 vCPU 60 GiB
1 * NVIDIA A10 -
2.9 GHz/3.5 GHz 16 Gbps 300 万 PPS
12 万/- 是 ¥ 13.458 /时
Intel Xeon(Ice Lake) Platinum 8369B
13.6元/小时
Ubuntu 20.04 64位
安装GPU驱动 CUDA版本11.2.2 Driver版本 460.91.03
CUDNN版本 8.1.1
40G的内存
pip3 install torch==1.10.0+cu113 torchvision==0.11.1+cu113 torchaudio==0.10.0+cu113 \ -f https://download.pytorch.org/whl/cu113/torch_stable.html
环境安装
python 3.6
pip3更新
pip3的源地址,加速下载
下载torch、torchvision和torchaudio,并离线安装。
数据准备
训练数据和测试数据
训练代码
测试代码
需要作图的数据
文章评论