작심삼일

[Pytorch 에러] OSError: libcudart.so.10.2: cannot open shared object file: No such file or directory 본문

정리/잡 에러들

[Pytorch 에러] OSError: libcudart.so.10.2: cannot open shared object file: No such file or directory

yun_s 2022. 5. 9. 14:26
728x90
반응형

에러

CUDA버전이 맞지 않을때 나타나는 에러다.

Pytorch를 재설치하면 된다.

 

CUDA 11.3 버전 설치

# conda
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

# pip
pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113

 

PyTorch

An open source machine learning framework that accelerates the path from research prototyping to production deployment.

pytorch.org

혹은 위 페이지에서 원하는 버전으로 선택하면 된다.

728x90
반응형
Comments