JEon.E
일단 ML 엔지니어 생존일지
JEon.E
전체 방문자
오늘
어제
  • 분류 전체보기 (164)
    • 논문 스터디 (8)
      • 논문 구현 (0)
      • Code-LLM (0)
      • ML attack (6)
      • Fuzzing (2)
    • 동향 및 조사 (3)
    • Stack Overflow (6)
    • Setting Tips (14)
    • ML 엔지니어링 (1)
      • AI Math (0)
      • Pytorch (1)
    • 알고리즘 (132)
      • 이론 (8)
      • 문제풀이 (105)
      • 삼성 기출 문제풀이 (18)
    • Hack (0)
      • 해킹 맛보기 (0)

블로그 메뉴

  • 홈

공지사항

인기 글

태그

  • 다시
  • 플로이드-워샬
  • Adversarial Defense
  • BFS
  • Graph
  • 백트래킹
  • 크루스칼
  • keras
  • Fuzzing 동향
  • 프로그래머스
  • 시뮬레이션
  • BOJ
  • dfs
  • dp
  • 강화학습
  • 그래프
  • 구현
  • NLP black-box attack
  • 네이버 부스트캠프
  • greedy

최근 댓글

최근 글

티스토리

hELLO · Designed By 정상우.
JEon.E
Setting Tips

아나콘다 tensorflow-gpu 가상환경 버전 맞추어 설치하기

Setting Tips

아나콘다 tensorflow-gpu 가상환경 버전 맞추어 설치하기

2021. 12. 3. 04:36

 

 

conda create -n 가상환경명 tensorflow-gpu명령어를 그대로 사용하면 한 가지 문제가 발생한다.

바로 conda 가상환경은 tensorflow-gpu와 함께 가상환경을 생성할 때, CUDA toolkit과 cuDNN을 함께 설치하지만 텐서플로우 버전과 상관없이 항상 최신 버전을 설치한다는 것.....

 

만일 tensorflow-gpu 버전을 지정한다면 버전 문제로 인하여 종종 gpu를 사용하지 못하는 일이 발생하게 된다. 

그렇다고 cuda toolkit과 cudnn을 가상환경이 아닌 로컬로 돌아가게 하기에는 환경변수 설정만 문제가 아니다. 다른 tensorflow 버전(특히 1.x버전)도 쓰기 굉장히 까다로워진다. 

필자도 해당 문제를 겪었고.....결국 conda 명령어로 수동으로 cudnn, cudatoolkit, tensorflow-gpu를 설치하여 해결하는 방법을 찾았다. 

 

해당 방법의 장점은 가상환경마다 tensorflow-gpu 버전을 바꾸어 사용할 수 있다는 점. 

만일 해당 문제로 골머리를 앓는 사람이 있었다면 속 시원하게 해결하고 가길 바란다. 

 

1. 가상환경 생성하기 

conda create -n 가상환경명 

 

 

2. 가상환경을 활성화한 후, 원하는 버전의 tensorflow-gpu를  conda 명령어로 설치한다.

원하는 버전이 anaconda에 없을 수도 있다. https://anaconda.org/anaconda/tensorflow-gpu/files 를 통해 anaconda가 지원하는 버전을 확인하자.

conda  install -c anaconda tensorflow-gpu=2.x (최신 버전일 경우, tensorflow-gpu만 작성하여도 무방함)

 

 

3. tensorflow-gpu 버전에 맞는 cudnn과 CUDA toolkit을 해당 페이지에서 찾는다. 

https://www.tensorflow.org/install/source#tested_build_configurations

 

 

소스에서 빌드  |  TensorFlow

Help protect the Great Barrier Reef with TensorFlow on Kaggle Join Challenge 소스에서 빌드 소스에서 TensorFlow pip 패키지를 빌드하고 Ubuntu Linux 및 macOS에 설치합니다. 명령어는 다른 시스템에도 적용될 수 있지만, U

www.tensorflow.org

 

4. 찾은 버전의 cudnn과 CUDA toolkit을 conda 명령어로 설치해주자. 

conda install -c conda-forge cudnn=8.x (최신 버전일 경우, cudnn만 작성해도 무방함)
conda install -c conda-forge cudatoolkit=10.x (최신 버전일 경우, cudatoolkit만 작성해도 무방함)

아나콘다가 모든 버전을 갖고 있는 건 아니라서 맞추기가 살짝 까다롭다. 아래 명령어로 한꺼번에 해결하길 추천함

create -n [가상환경 명] python=3.8 tensorflow-gpu=2.2.0 cudatoolkit=10.1 cudnn=7.6 

반응형
저작자표시 비영리 변경금지 (새창열림)

'Setting Tips' 카테고리의 다른 글

vscode liveshare로 원격 코딩하기  (0) 2022.04.20
Conda 가상환경 설정 옮길 때 팁  (0) 2022.04.14
공유기 IP 고정하기  (0) 2022.03.28
외부에서 서버와 SSH를 통해 visual studio code 원격 연결하기  (0) 2022.01.30
크롬 원격 데스크탑 & 아나콘다 환경 구축(ubuntu 20.04 기준)  (0) 2021.07.19
    'Setting Tips' 카테고리의 다른 글
    • Conda 가상환경 설정 옮길 때 팁
    • 공유기 IP 고정하기
    • 외부에서 서버와 SSH를 통해 visual studio code 원격 연결하기
    • 크롬 원격 데스크탑 & 아나콘다 환경 구축(ubuntu 20.04 기준)
    JEon.E
    JEon.E
    ML Security Engineer로 살아남기 도전중

    티스토리툴바

    개인정보

    • 티스토리 홈
    • 포럼
    • 로그인

    단축키

    내 블로그

    내 블로그 - 관리자 홈 전환
    Q
    Q
    새 글 쓰기
    W
    W

    블로그 게시글

    글 수정 (권한 있는 경우)
    E
    E
    댓글 영역으로 이동
    C
    C

    모든 영역

    이 페이지의 URL 복사
    S
    S
    맨 위로 이동
    T
    T
    티스토리 홈 이동
    H
    H
    단축키 안내
    Shift + /
    ⇧ + /

    * 단축키는 한글/영문 대소문자로 이용 가능하며, 티스토리 기본 도메인에서만 동작합니다.