在多GPU的机器上,指定使用某(几)个GPU运行CUDA相关的程序:
CUDA_VISIBLE_DEVICES=0,1 python train_cnn.py
上一篇 Ubuntu16.04 CUDA与OpenGL分离的环境配置
下一篇 Ubuntu 使用Python HTTP Server共享文件