[code issue] torch GPU 버전 설치 에러

torch GPU 버전을 활용하려고 하다가 에러가 났다. 

 

huggingface transformer 최신 라이브러리가 pytorch 2.6 미만 버전은 보안 이슈로 강제로 차단을 하고 있기 때문에 버전을 낮춰줘야 한다. 

 

(물론 여기서 cuda toolkit의 버전을 확인해야하는데, cuda toolkit 버전이 12.6 이상이면 torch 버전 2.8 이상을 쓸 수 있다. 그럼 GPU도 그에 맞춰서 버전을 더 높은거를 맞춰주면 된다. 나의 경우는 cuda가 12.1인가 낮은 버전이었기 때문에, torch gpu 버전이 2.6 미만을 활용헤야 했다. 따라서, 본인의 환경을 확인하여 맞게 설치해야한다!!)

 

pip install torch==2.5.1+cu121 --index-url https://download.pytorch.org/whl/cu121

 

해당 코드를 통해서 다운 그레이드를 진행해주면 된다.

 

나의 경우 poetry에서 실행을 했는데, 직접적인 설치가 불가능(?) 했기 때문에 다음과 같이 설치해주었다.

 

poetry run pip install torch==2.5.1+cu121 --index-url https://download.pytorch.org/whl/cu121

 

이렇게 해서 오늘의 이슈 사항 정리 해보았다.

 

감사합니다.