pytorch hook feature map 출력하기

register_forward_hook 을 이용하여 torchvison의 vgg19 모델의 feature map에 대한 연산을 수행합니다.

더 보기

ubuntu 16.04 Pytorch 설치(pip, source)

2018 8.14 일 업데이트 nvidia drive,cuda,cudnn버젼을 최신버젼에 맞게 변경 1. 그래픽 드라이버 설치 sudo add-apt-repository ppa:graphics-drivers/ppa sudo apt-get update sudo apt-get install nvidia-396 설치 후  재부팅을 한다. 2. CUDA 9.2설치 pip으로 설치 시에는 9.2을 설치해야한다!(2018.8.14일 기준) 공식페이지에서 cuda설치 linux 16.04  deb(network) 다운로드 sudo dpkg -i cuda-repo-ubuntu1604_9.2.148-1_amd64.deb sudo apt-key adv –fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/7fa2af80.pub sudo apt-get update sudo apt-get…

더 보기