아하

생활

생활꿀팁

대범한바다표범103

대범한바다표범103

tensorflow_gpu 가상환경에 다운로드 2

아나콘다 가상환경에 텐서플로 gpu 버전을 설치를 하는데

위는 import tensorflow 커맨드 라인의 결과입니다. 그래서 CUDA file에 가봤는데

폴더가 비어 있길래 제가 다시 install 을 했는데 다운로드가 완료되고 몇 초 뒤면 알아서 삭제가 되어버립니다..

그래서 환경변수 때문인가 예상을 했습니다. cudnn을 바탕화면에 풀어서 환경변수 자체를 그쪽으로 넘겨줬는데

gpu 가상환경 안에서 동작이 되지 않습니다...혹시나 해서 cuda 9.0 , cudnn 7.4.2 버전으로 사용했습니다 (윈도우 10)현재 그래픽카드는 GTX 1050 입니다. 뭐가 문제인지 알려주세요 ㅠ

1개의 답변이 있어요!

  • 빠른너구리257

    빠른너구리257

    안녕하세요

    우선 환경변수를 잘못 잡아주신것 같습니다.

    쿠다를 위해 필요한 환경변수는 다음과 같습니다:

    • C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin

    • C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\libnvvp

    • C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\extras\CUPTI\libx64

    다음은 cudnn을 위한 환경변수위치 입니다.

    • C:\{cudnn 위치}\cuda\bin

    감사합니다