RTX 4090을 사용하려고 하는데 파이토치 CUDA 버전을 모르겠어요

4090 CUDA 버전이 11.8이던데 아직 파이토치는 11.7까지만 지원하는 것 같더라고요 찾아보니까 아직 compatible 하지 않다고 hungingface에 댓글로 적혀있던데 4090 을 사면 pytorch를 사용 못할까요? 그렇다고 3090을 사기엔 너무 아쉬워서 여쭤봅니다,!

아래 글을 참조해보면, nvidia에서 제공하는 container에
cuda 11.8 +pytorch 1.13 으로 환경 구성이 되어있습니다.

도커 사용하신다면, pytorch 사용은 가능할 듯 합니다(베타 버전)…
앞으로도 여러 테스트 중, 환경관련 문제로 호환이 맞지않아 골치 아플 수 있을 듯 하여
저라면 3090이나 TITAN쪽을 고려해볼 듯 합니다.

2개의 좋아요

넵 감사합니다. 잘 생각 해보겠습니다.

현재 4090 사용중입니다. 파이토치 사용은 가능합니다만 cuFFT 지원이 안되는 등 문제가 발생하는 경우가 있어요. torch.stft 같은거를 쓰려다가 실패해서 nvidia container로 사용중입니다.

1개의 좋아요