pytorch (2) 썸네일형 리스트형 timm 사용가능한 pretrained model weights 리스트 timm (PyTorch Image Models) 에서는 사전 학습된 model weights를 쉽게 다운받아서 활용할 수 있다. 이때 어떤 Pretrained model weights가 사용가능하고, 원하는 Weights를 사용하기 위해서 어떤 keyword를 사용해야 하는지를 간단하게 적어본다. 먼저, 사용가능한 Pretrained model weights 리스트를 뽑기 위해서는 아래와 같이 입력해주면 된다. import timm print(timm.list_models(pretrained=True)) 모델은 아래와 같이 생성할 수 있다. model = timm.create_model('efficientnet_b0.ra_in1k', pretrained=True) 자세한 내용은 공식 Documentat.. Torch Cuda compiler version mismatch Anaconda를 이용하여 아래와 같이 torch, cudatoolkit, cudnn 등을 설치하였다. conda install pytorch==1.13.1 torchvision==0.14.1 torchaudio==0.13.1 pytorch-cuda=11.7 -c pytorch -c nvidia 그리고 나서 Torch에서 cuda로 작성된 custom module을 컴파일하려고 하니 다음과 같은 에러가 발생하였다. The detected CUDA version (12.0) mismatches the version that was used to compile PyTorch (11.7). Please make sure to use the same CUDA versions Anaconda에서 cuda 관련된 .. 이전 1 다음