1. 设置多GPU环境下GPU的可见性

如果服务器上有多个GPU,可以设置程序只用某几个GPU。

export CUDA_VISIBLE_DEVICES=0,1
export CUDA_VISIBLE_DEVICES=1

在别人使用了默认的GPU0时,这个参数尤其有用。

Reference

https://stackoverflow.com/questions/39649102/how-do-i-select-which-gpu-to-run-a-job-on


文章版权归 FindHao 所有丨本站默认采用CC-BY-NC-SA 4.0协议进行授权|
转载必须包含本声明,并以超链接形式注明作者 FindHao 和本文原始地址:
https://findhao.net/easycoding/2595.html

Comments