본문 바로가기
  • AI 개발자가 될래요

pytorch2

[PyTorch/Python] conda list의 torch 버전과 pip show torch 버전이 다를 때 [PyTorch/Python] conda list의 torch 버전과 pip show torch 버전이 다를 때 pip show torch에 출력된 파이토치 버전과 conda list로 출력된 파이토치 버전이 다를 때 해결법이다. 1. 문제 상황파이토치를 설치했고, conda list 로도 출력이 되는데 no module name 'torch' 에러가 자꾸 떴다. 1.1 pip torch 버전 확인pip show torch위의 명령어로 torch 버전을 확인해보니 아래처럼 나왔다.Name: torchVersion: 2.2.1Summary: Tensors and Dynamic neural networks in Python with strong GPU accelerationHome-page: https://.. 2025. 2. 21.
[pytorch/분산처리/디버깅노트] 배치사이즈 주의점 / 배치사이즈가 1일 때 분산처리 주의사항 / 에러 원인 및 해결방법 [pytorch/분산처리] 배치사이즈 주의점 / 배치사이즈가 1일 때 분산처리 주의사항 / 에러 원인 및 해결 방법 인고의 디버깅 끝에 알아낸 문제와 해결 과정 및 방법. 1. 문제현재 어떤 오픈소스 모델 학습 중에 있는데, 건드린게 없는데도 학습 시 에러가 발생했다.audio = audio.view(-1, 5, audio.shape[-1]) # [B, T, 128]RuntimeError: shape '[-1, 5, 128]' is invalid for input of size 256 해석하자면 audio 변수를 [b, 5, 128] 형태로 바꿀 수 없다는 것이다.audio 변수는 코드상으로는 [b*5, 128] 형태이기 때문에, [b, 5, 128]으로 당연히 바꿀 수 있는 것이었다. 그러나 특정 구간.. 2025. 2. 18.

티스토리툴바