安装过程掉了好多坑,看到这篇安装教程觉得直白容易上手,而且亲测有效,于是就翻译一下,并根据实际一般情况进行了修改。
原文链接:https://millansanchez.medium.com
安装之前的准备:
- 确保电脑安装了NVIDIA驱动程序,且计算能力在3.5以上(CUDA GPUs可以查询对应型号的计算能力);
- 安装Anaconda或Miniconda。
安装过程:
- 先安装Jupyter Notebooks(安装Anaconda的可跳过)
conda install -y jupyter
2. 为 Python 3.7 和 TensorFlow 创建一个虚拟环境(TensorFlow 目前支持 Python 3.8,不使用是因为后面无法进行自动安装)
conda create --name tensorflow-gpu python=3.7
3. 激活环境(后续命令也在这个环境中运行)
activate tensorflow-gpu
4. 在此环境中安装 Jupyter Notebooks 支持,并且同意安装所需的组件
conda install nb_conda
5. 安装tensorflow-gp