| 일 | 월 | 화 | 수 | 목 | 금 | 토 |
|---|---|---|---|---|---|---|
| 1 | 2 | 3 | 4 | |||
| 5 | 6 | 7 | 8 | 9 | 10 | 11 |
| 12 | 13 | 14 | 15 | 16 | 17 | 18 |
| 19 | 20 | 21 | 22 | 23 | 24 | 25 |
| 26 | 27 | 28 | 29 | 30 | 31 |
- paramiko
- pandas
- 채보
- error
- SSH
- YOLO
- Linux
- VS Code
- 기타 연주
- C#
- 오류
- JSON
- Numpy
- label
- mysql
- Python
- pytorch
- 컨테이너
- Docker
- ubuntu
- C++
- 핑거스타일
- pip
- Selenium
- Visual Studio
- LIST
- 프로그래머스
- windows forms
- OpenCV
- C
- Today
- Total
목록torch (2)
기계는 거짓말하지 않는다
Python torch 프레임워크 실행 시 다음과 같은 오류가 발생한 경우Could not load library libcudnn_cnn_train.so.8. Error: /usr/local/cuda-12.2/lib64/libcudnn_cnn_train.so.8: undefined symbol: _ZN5cudnn3cnn34layerNormFwd_execute_internal_implERKNS_7backend11VariantPackEP11CUstream_stRNS0_18LayerNormFwdParamsERKNS1_20NormForwardOperationEmb, version libcudnn_cnn_infer.so.8Traceback (most recent call last): File "/python_p..
torch의 load를 이용하여 model weight를 불러올 때, module. model 등이 key 값에 더 붙어있거나, key 이름이 다른 경우 변경하여 가지고 올 수 있다. 단, model 구조는 같아야 한다. state_dict = checkpoint[state_key] new_state_dict = {} # load된 model의 key에 model. 이 붙어있을 경우 제거 for k, v in state_dict.items(): if "model." in k: name = k[6:] new_state_dict[name] = v print(new_state_dict.keys()) if len(new_state_dict.keys()) == 0: model.load_state_dict(stat..
