pytorch2 tensor 내용 전체 출력 및 파일 저장 말 그대로다. pytorch에서 tensor로 저장되는 weight 값을 저장하고 싶었다. 그것도 줄임표시가 없이 전체 저장!! print를 하면 ... 표시로 나오고 파일로 저장해도 ... 로 나오고.. 아래 글처럼, numpy 배열을 뿌리는 옵션을 찾았지만 tensor에는 적용되지 않았다. 구글링으로 set_printoptions 를 찾았고, 아래 numpy 옵션 처럼, 라인 등 지정이 가능하다. 파일로 저장하는 부분도 함께 찾았는데, 자세한 부분은 아래 링크 참조하여 상황에 맞게 수정하여 사용. import pandas as pd # profile 부분에서 설정 가능 # full : 전체 출력 설정 torch.set_printoptions(profile="full") state_dict = mode.. 2023. 4. 12. AssertionError: Torch not compiled with CUDA enabled 파이 토치 + CUDA(v11.3) 설치 후 CMD 창에서 직접 파이썬을 호출하여 CUDA 사용 여부를 확인해보면 True를 뱉어내는데.. 꼭, 주피터 노트북에서 CUDA 사용 여부를 다시 확인하면 계속 False를 뱉어내고, GPU 장치 정보를 가져오려고 하면 "Torch not compiled with CUDA enabled" 메시지가 나온다. 계속 가상환경을 지웠다 만들었다가... 파이토치도 지웠다 다시 설치했다를 반복했다가... 에효,, 암튼 계속되는 구글링... 결국 해결한 방법은 Conda 설치가 아닌 Wheel 방식(pip) 설치. 아래 명령어로 파이토치랑 CUDA 버전 지정해서 설치함. pip3 install torch==1.10.0+cu113 torchvision==0.11.1+cu113.. 2022. 4. 6. 이전 1 다음
