alias - 如何使用别名来简化 CUDA_VISIBLE_DEVICES

Question

我可以alias gpu0='CUDA_VISIBLE_DEVICES=0'用来设置gpu0，但是如果CUDA_VISIBLE_DEVICES=0,1,2呢？

标签： aliasgit-bash

Answer 1

我为此提出了一个bash函数。将此函数放入 .bashrc 中：

gpu() {
  export CUDA_VISIBLE_DEVICES="$1"
}

并将其与gpu 0or一起使用gpu 0,1