티스토리 뷰
FROM pytorch/pytorch:2.6.0-cuda12.6-cudnn9-devel
# CUDA 라이브러리 경로 환경변수 설정 for onnxruntime-gpu
ENV LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
ENV LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/opt/conda/lib/python3.11/site-packages/nvidia/cudnn/lib
Dockerfile혹은 안에 컨테이너에서 위와 같이 LD LIB PAHT에 저 두 경로를 추가해주자
cudnn 라이브러이 위치들이 이누야샤 사원의 조각마냥 흩뿌려져있어서 lib경로 다 찾아서 추가해줬다.
'Deep Learning' 카테고리의 다른 글
| u-net이 이리 오래전에 나온 모델일줄이야 (0) | 2025.04.06 |
|---|---|
| 2025년 2월 22일 기준 최신 vision encoder! (0) | 2025.02.22 |
| 다양한 최신 백본으로 Image Classification 성능 측정한 논문 - 다시 보기용 (0) | 2025.02.19 |
| torch2onnx transformer encoder, decoder onnx변환시 aten::_native_multi_head_attention 에러 우회법 (0) | 2025.02.18 |
| face recognition task와 face alignment (0) | 2025.01.31 |
댓글
공지사항
최근에 올라온 글
최근에 달린 댓글
- Total
- Today
- Yesterday
링크
TAG
- PyCharm
- 백준 11437
- cosine
- 위상 정렬 알고리즘
- ㅂ
- FairMOT
- 백준 1766
- 백트래킹
- Lowest Common Ancestor
- 백준
- 순열
- 파이참
- 문제집
- LCA
- 가장 긴 증가하는 부분 수열
- 단축키
- 이분탐색
- 자료구조
- MOT
- 백준 11053
- 인공지능을 위한 선형대수
- 조합
- C++ Deploy
| 일 | 월 | 화 | 수 | 목 | 금 | 토 |
|---|---|---|---|---|---|---|
| 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 |
글 보관함