728x90
python version = 3.8
pycharm에서는 cuda가 잡히지만 jupyter에서는 잡히지 않는 현상이 발생함
가상환경을 맞춰줬는데도 이런 현상이 발생하였음
import os
#os.environ['CUDA_VISIBLE_DEVICES'] = '1'
import numpy as np
import math
from PIL import Image
import matplotlib.pyplot as plt
from tqdm.notebook import tqdm
os.environ['CUDA_VISIBLE_DEVICES'] = '1'
문제는 바로 이 코드! 나중에 서버컴에서 돌리려고 넣어둔 코드인데 Devices가 없다고 인식하여 False가 나온 것 같다.
728x90