cpu
pip install tensorflow==2.2.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install numpy pandas matplotlib sklearn -i https://pypi.tuna.tsinghua.edu.cn/simple
Anaconda最新版本的下载地址
清华大学镜像地址
https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/
中科大镜像地址
http://mirrors.ustc.edu.cn/anaconda/archive/
GPU
安装GPU版本必须有GPU 硬件的支持。TensorFlow 对 NVIDIA 显卡的支持较为完备。我们使用 conda 来安装GPU版本
对于 NVIDIA 显卡,要求其 CUDA Compute Capability 须不低于 3.0。算力参考: https://developer.nvidia.com/cuda-gpus
驱动版本注意:NVIDIA驱动程序需 410.x 或更高版本。
以下两个是依赖库的版本要求,不需要手动安装。
CUDA的版本需要是 10.0
cudnn版本号需要不小于 7.4.1
conda install cudnn=7.6.0
conda install cudatoolkit=10.0.130
NVIDIA驱动程序下载地址
https://www.nvidia.com/Download/index.aspx?lang=en-us
Visual C++ 安装,可从微软的官网下载
https://support.microsoft.com/zh-cn/help/2977003/the-latest-supported-visual-c-downloads
安装GPU版本:
conda install tensorflow-gpu==2.0.0