通常实验室的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
通常实验室的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