【linux备忘录】配置可使用的gpu资源

Ra1ny House bio photo By Ra1ny House

通常实验室的gpu资源有限,而tensorflow默认使用gpu0,会造成冲突。如何设置只使用部分gpu资源呢,使用下列命令即可:

CUDA_VISIBLE_DEVICES=1 python yourfile.py

或者在.zshrc(.bash_profile)文件中添加

# 配置可使用的gpu, eg CUDA_VISIBLE_DEVICES=0,1
export CUDA_VISIBLE_DEVICES=1