Azure N系 Ubuntu 环境配置

131 阅读1分钟

踩了很多坑,官方文档只介绍到install cuda,不够靠谱。

找到一份standford cs234 azure 配置脚本,一气呵成。

首先,在azure官网配置Ubuntu16.04虚拟机,我选择的是Standard_NC6。

git clone https://github.com/AndyYSWoo/Azure-GPU-Setup.git
cd Azure-GPU-Setup
chmod +x *.sh
./gpu-setup-part1.sh
# reboot
nvidia-smi
cd Azure-GPU-Setup
./gpu-setup-part2.sh
python gpu-test.py

如上操作之后,Ubuntu16.04 + CUDA9.0 + cudnn7.0 就配好了。