Привет ребята и девушки,
В этом руководстве мы установим графический процессор 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 | ТензорФлоу
- Скачать версию CUDA 11.0.x для Windows 10 (я выбираю cuda_11.0.3_451.82_win10.exe)
- Установите, следуя инструкциям. (Продолжайте нажимать Далее :)))
- Добавьте путь к переменным среды:
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 и играй :)))
- Создайте среду 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