728x90
반응형
https://ssaru.github.io/2021/05/05/20210505-til_install_rtx3090_supported_pytorch/
(TIL) RTX 3090을 지원하는 PyTorch 버전설치
2021.05.05 현재 RTX3090은 CUDA11 이상을 지원하는 딥러닝 프레임워크에 버전에서만 사용할 수 있습니다. 하지만 단순하게 pip install torch==1.7.1 torchvision==0.8.2 형태로 설치하면 CUDA error: no kernel image is ava
ssaru.github.io
pip install torch==[버전]
이런식으로 설치하면
CUDA error: no kernel image is available for execution on the device
계속 이 에러가 떠서 설치할 수 없음
pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 -f https://download.pytorch.org/whl/torch_stable.html
이 포맷으로 설치해줘야 함
728x90
반응형
'사소한 Tip . 오류 해결법' 카테고리의 다른 글
ImportError: IProgress not found (0) | 2022.02.22 |
---|---|
ValueError: transformers.__spec__ is None (0) | 2022.02.21 |
AttributeError: module 'distutils' has no attribute 'version' (0) | 2022.02.21 |
마크다운/ HTML 표 생성 (0) | 2022.02.04 |
[Jupyter Notebook, HTML] 주피터 노트북에서 HTML 읽고 화면 출력하기 / How to display HTML contents in Jupyter Notebook (0) | 2022.01.14 |