tensorflow安装

通过annaconda来安装TensorFlow

annaconda安装方式:查看我另外一篇博客【点击跳转

新建tensorflow环境

因为不同的框架公用的部分工具会存在兼容性问题,所以在annaconda里面,最好新建一个TensorFlow单独的环境:

# tf为环境名称,自定义
# 3.6是指定python版本,根据自己需要设置所需版本号
conda create -n tf python=3.6

切换到TensorFlow环境

conda active tf

tensorflow安装:

使用pip连接清华源进行安装,速度会比较快

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow

CUDA警告

警告如下:

2021-03-12 15:02:54.791342: W tensorflow/stream_executor/platform/default/dso_loader.cc:60] Could not load dynamic library 'cudart64_110.dll'; dlerror: cudart64_110.dll not found
2021-03-12 15:02:54.791414: I tensorflow/stream_executor/cuda/cudart_stub.cc:29] Ignore above cudart dlerror if you do not have a GPU set up on your machine.

解决办法参考链接

安装完成后,在shell输入python命令,进入环境

import tensorflow

解决jupyter中无法使用tensorflow问题

依次输入如下命令:

pip install jupyter
pip install ipython
ipython kernelspec install-self --user

参考链接