반응형
Video editing model을 돌리다가 아래와 같은 에러가 conv2d에서 발생함을 확인했다 ㅋㅋ
File "/mnt/petrelfs/zhaozhiyuan/anaconda3/envs/minigpt4-nightly/lib/python3.9/site-packages/torch/nn/modules/conv.py", line 459, in _conv_forward
return F.conv2d(input, weight, bias, self.stride,
RuntimeError: GET was unable to find an engine to execute this computation여러가지 서치를 해본결과 뭐 cuda version이 안맞아서 그런거라는데.. 됐고
그냥 아래와 같이 추가해주면 된다. 11.8 부분에는 자신의 cuda version을 넣어주면 된다.
$ export PATH=/usr/local/cuda-11.8/bin${PATH:+:${PATH}}
$ export LD_LIBRARY_PATH=/usr/local/cuda-11.8/lib64
출처는 이 곳이다!
https://github.com/pytorch/pytorch/issues/102535#issuecomment-1666535370
https://forums.developer.nvidia.com/t/path-ld-library-path/48080
PATH & LD_LIBRARY_PATH
Set up the development environment by modifying the PATH and LD_LIBRARY_PATH variables: $ export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}} $ export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64\ {LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}} The above paragraph is
forums.developer.nvidia.com
반응형
'Linux' 카테고리의 다른 글
| [TIL] torch.distributed.DistBackendError: NCCL error, Internal check failed 에러 해결하기 (0) | 2025.02.05 |
|---|---|
| [Linux] Huggingface model default 저장 경로 (./.cache) 변경하기 (HF_HOME 지정) (1) | 2024.09.30 |
| [Git] 특정 브랜치만 clone하기 (0) | 2024.03.05 |
| [Linux] 갑자기 conda를 찾을 수 없을 때 ~/.bashrc 확인하기 (2) | 2024.01.06 |
| [Linux] .cache 폴더 삭제하기 - conda, pip cache 파일 삭제하기 (1) | 2024.01.06 |