반응형
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
CUDA_VISIBLE_DEVICES를 os environ 함수로 지정해주면 된다.
반응형
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
CUDA_VISIBLE_DEVICES를 os environ 함수로 지정해주면 된다.