티스토리 뷰

gist.github.com/Mahedi-61/2a2f1579d4271717d421065168ce6a73

 

CUDA 10.1 Installation on Ubuntu 18.04

CUDA 10.1 Installation on Ubuntu 18.04. GitHub Gist: instantly share code, notes, and snippets.

gist.github.com

 

위 깃허브를 따라하면 됩니다.  !!! 단 cudnn 설치할때는 NDVIA에 로그인을 해야합니다.

 

저는 크롬으로 로그인을 하였는데, 어떻게 연동시키는지 몰라 직접 다운 받았습니다. (연동시키지 않으면 아래 코드가 실행이 안됩니다.)

wget https://developer.nvidia.com/compute/machine-learning/cudnn/secure/7.6.5.32/Production/10.1_20191031/cudnn-10.1-linux-x64-v7.6.5.32.tgz

developer.nvidia.com/rdp/cudnn-archive

 

cuDNN Archive

NVIDIA cuDNN is a GPU-accelerated library of primitives for deep neural networks.

developer.nvidia.com

위 사이트에서 로그인을 하면, cudnn을 다운받을 수 있습니다. 여기서 10.1에 맞는 7.6.5버전을 다운받으면 됩니다.

 

맨 마지막 줄에있는 코드는 텐서플로우를 설치하는 코드입니다.

댓글
최근에 올라온 글
최근에 달린 댓글
250x250