먼저 내 환경에 대해 이야기하겠습니다.
쿠다==11.2
파이토치==1.12.1
텐서플로우==2.6.0
1. 먼저 실행중인 cuda 버전을 확인하고,
nvcc --version cuda 버전 보기
11.2로 정상으로 나타났습니다.
2. 텐서플로 보기
Tensorflow를 tf로 가져오기
tf.test.is_gpu_aviauable()
반환된 true는 환경이 정상임을 나타냅니다.
3. 마지막으로 pytorch 버전에서 문제를 찾아 버전을 변경하고 버전 1.8.1로 업그레이드합니다.구체적인 이유는 1.12.1이 11.2의 cuda 버전을 지원하지 않기 때문이라고 생각합니다.