The quick way to make the model inferences or fine-tuning running on specific NVIDIA GPU card is by define this variable before execute the script
For instance, I forced its running on GPU 1 by
export CUDA_VISIBLE_DEVICES=1
Categories
The quick way to make the model inferences or fine-tuning running on specific NVIDIA GPU card is by define this variable before execute the script
For instance, I forced its running on GPU 1 by
export CUDA_VISIBLE_DEVICES=1