TensorFlow 准备 JupyterLab 交互式笔记本环境,方便我们边写代码、边做笔记。
基础环境
以下是本文的基础环境,不详述安装过程了。
Ubuntu
- Ubuntu 18.04.5 LTS (Bionic Beaver)
- ubuntu-18.04.5-desktop-amd64.iso
CUDA
- CUDA 11.2.2
- cuda_11.2.2_460.32.03_linux.run
- cuDNN 8.1.1
- libcudnn8_8.1.1.33-1+cuda11.2_amd64.deb
- libcudnn8-dev_8.1.1.33-1+cuda11.2_amd64.deb
- libcudnn8-samples_8.1.1.33-1+cuda11.2_amd64.deb
Anaconda
- Anaconda Python 3.8
- Anaconda3-2020.11-Linux-x86_64.sh
conda activate base
安装 JupyterLab
Anaconda 环境里已有,如下查看版本:
jupyter --version
不然,如下进行安装:
conda install -c conda-forge jupyterlab
安装 TensorFlow
创建虚拟环境 tf
,再 pip
安装 TensorFlow:
# create virtual environment
conda create -n tf python=3.8 -y
conda activate tf