[Cell2Location] Cell2Location 구형 CUDA 드라이버에서 돌리는 법

현재 사용중인 서버는 CUDA 11.4 버전이다.

root 가 나한테 없어서, 드라이버를 바꾸긴 어렵다. 근데 Cell2Location 은 Pytorch 2.x 로 되어있고, CUDA 12 로 맞게 만들어져 있어서 cell2location 기본 설치로 하면 RuntimeError가 뜨면서 학습이 안된다.

이럴 경우 어떻게 해결하는가

우선 CUDA 를 해결해주기 전에, scvi 쪽에서 문제가 생긴다.

이 부분은 특정 패키지들을 다운그레이드해주면 해결되는 부분.

요것으로 해결가능했다.

그렇다면 CUDA 12가 아닌 11.4 에 맞는 파이토치와 그 친구들로 바꿔줘야 한다.

그러기 위해서

# CUDA 11.8
conda install pytorch==2.2.0 torchvision==0.17.0 torchaudio==2.2.0 pytorch-cuda=11.8 -c pytorch -c nvidia

의 코드로 pytorch 버전을 바꿔주면

해결된다!