This commit is contained in:
Guo Liyong 2022-05-27 10:37:29 +08:00
parent d9a6aff863
commit aa9e287b3a

View File

@ -9,7 +9,13 @@ stage=$1
# Set the GPUs available. # Set the GPUs available.
# This script requires at least one GPU. # This script requires at least one GPU.
# Example: # You MUST set environment variable "CUDA_VISIBLE_DEVICES",
# even you only have ONE GPU. It needed by CodebookIndexExtractor to determine numbert of jobs to extract codebook indexes parallelly.
# Suppose only one GPU exists:
# export CUDA_VISIBLE_DEVICES="0"
#
# Suppose GPU 2,3,4,5 are available.
export CUDA_VISIBLE_DEVICES="2,3,4,5" export CUDA_VISIBLE_DEVICES="2,3,4,5"