Привет ребята и девушки,

В этом руководстве мы установим графический процессор Tensorflow с Anaconda Environment в Pycharm. Приятного чтения!

Во-первых, давайте проверим мою систему:

  • Операционные системы: Windows 10
  • Версия Pycharm: 2021.1.3
  • Графический процессор: NVIDIA GeForce GTX 1650 Ti.

Вы можете проверить информацию о своем графическом процессоре, нажав «Windows + X», затем введите «dxdiag».

Установка CUDA и Cudnn.

Это самая важная часть! ЖДИТЕ и будьте внимательны :))

Вы можете проверить совместимую версию CUDA для Tensorflow через Build from source | ТензорФлоу

  1. Скачать версию CUDA 11.0.x для Windows 10 (я выбираю cuda_11.0.3_451.82_win10.exe)
  2. Установите, следуя инструкциям. (Продолжайте нажимать Далее :)))
  3. Добавьте путь к переменным среды:
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\libnvvp

4. Скачать cudnn-11.2-windows-x64-v8.1.1.33.zip и распаковать.

5. Скопируйте эти файлы из извлеченной папки cudnn в CUDA:

1. cuda\bin\cudnn64_8.dll to C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\bin\ 
2. cuda\include\cudnn.h to C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\include\
 
3. cuda\lib\x64\cudnn.lib to  C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\lib\x64\

Открой Pycharm и играй :)))

  1. Создайте среду Anaconda в Pycharm
File > Settings >Project> Python Interpreter

2. Добавьте пакет Tensorflow GPU 2.5 (или 2.4)

File > Settings >Project> Python Interpreter

ВЫПОЛНЕНО!

Давайте проверим вашу установку:

import tensorflow as tf 
import os os.environ["CUDA_VISIBLE_DEVICES"] = '0' print(tf.__version__)

В случае успешной установки терминал вернет:

Всякий раз, когда вы тренируете модель, которая назначает графический процессор Tensorflow, вы получите подробное сообщение:

Рекомендация:

Вы можете изменить свой терминал Pycharm с локального на Conda:

File > Settings> Tools> Terminal:
Change shell path: 
cmd.exe "/K" "C:\Users\YOUR_NAME\anaconda3\Scripts\activate.bat"

Обновления ошибок:

Если вы встретите какие-либо ошибки, пожалуйста, прокомментируйте этот пост! Мы можем попытаться решить это вместе :D

Приятного обучения!