Tensorflow
- 查看与python/cuda/cudnn的版本对应,根据要安装的版本,切换或安装所需cuda版本。
注意:python3.6兼容所有tensorflow版本 - 为TF建立虚拟环境(推荐)
- 激活新建的虚拟环境。安装依赖项及TF:
pip install -U pip numpy wheel pip install -U keras_preprocessing --no-deps pip install tensorflow-gpu==1.15
- 注意:若要安装cpu版,可在此处下载对应支持包,使用以下命令安装:
否则在电脑支持的情况下,会出现类似下面的提示,表示使用pip install --ignore-installed --upgrade "Download URL"
AVX2 FMA
编译会获得更好的效果(但仍远不如gpu)Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2 FMA
- 注意:若要安装cpu版,可在此处下载对应支持包,使用以下命令安装:
- 验证安装成功(应返回
Tensor("Sum:0", shape=(), dtype=float32)
)python -c "import tensorflow as tf;print(tf.reduce_sum(tf.random.normal([1000, 1000])))"