Alvin's Blog 这里记录着我的学习之路

CUDA 指定GPU

2017-11-03
Alvin Zhu

在多GPU的机器上,指定使用某(几)个GPU运行CUDA相关的程序:

CUDA_VISIBLE_DEVICES=0,1 python train_cnn.py

相关文章

评论