[Pytoch] torchvision pretrain 가중치 다운로드 시 PermissionError: [Errno 13] Permission denied: 오류 해결

es.Seong·3일 전
0

삽질정리

목록 보기
5/5

문제발생

GPU 서버 이용 시 전이학습을 위해 torchvision.models에서 제공하는 가중치를 다운받다가 오류가 발생했다. 에러 로그는 아래 이미지와 같았다.

PermissionError: [Errno 13] Permission denied: '/.cache'

다운로드되는 가중치가 저기에 저장되는 것 같은데 아마 내 계정의 권한은 해당 폴더에 접근권한이 없어서 발생하는 오류였다.

해결

그렇다면 다운로드 경로가 설정된 환경변수 값 (TORCH_HOME)을 바꿔주면된다.
export를 사용해서 바꿔도되지만 코드로 설정값을 바꿔줬다.

import os
os.environ["TORCH_HOME"] = "/원하는경로"

이렇게 설정 후 실행하면

원하는 경로에 pth 가중치 파일이 잘 다운로드 된 것을 확인할 수 있다.

profile
Graduate student at Pusan National University, majoring in Artificial Intelligence

0개의 댓글

관련 채용 정보