租用阿里云服务器做AI训练

2021年 12月 9日 84点热度 0人点赞

 

服务器选型

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,并离线安装。

数据准备

训练数据和测试数据

训练代码

测试代码

需要作图的数据

 

rainbow

这个人很懒,什么都没留下

文章评论