TensorFlow安装
date
May 7, 2022
Property
slug
有点麻烦
status
Published
tags
工程经验
summary
type
Post
平时基本用Pytorch比较多，tensorflow听之前老师说比较适用于工程？？不过tensorflow1.X和tensorflow2.X兼容性不好确实挺麻烦的～。
Cuda10.2安装tensorflow1.X
使用conda安装，第一次仔细看发现conda自己原来会处理cudnn，所以就是在后面使用docker时，其实只要关注其中的系统版本和cuda版本即可，cudnn可以不用理他，cudnn就只是一个运行库而以。
并且conda会自己匹配对应的cudatoolkit，即使是cuda10.2但是conda其依然会自动匹配到cudatoolkit==10.1，并依然可以正常运行。
- 创建环境
conda create --name tensorflow_gpu python=3.6- 激活环境
activate tensorflow_gpu- 直接安装
conda install tensorflow-gpu==1.14.0- 安装对应的keras
pip install keras==2.2.5- 安装对应的常用库
pip install Pillow
pip install matplotlib
pip install opencv-python
pip install xlwt
pip install PyQt5- 查看是否可以正常使用cuda运算：总是会报各种warning，不过是正常的～
import tensorflow as tf
print(tf.test.is_gpu_available())