环境设置
base
python 3.9
jupyter 1.0.0
jupyter_client 7.3.4
jupyter_console 6.4.3
jupyter_core 4.11.1
jupyter_server 1.18.1
jupyterlab 3.4.4
jupyterlab_pygments 0.1.2
jupyterlab_server 2.10.3
jupyterlab_widgets 1.0.0
ipykernel 6.15.2
ipython 7.31.1
nb_conda_kernels 2.3.1
tf
python 3.9
tensorflow 2.6.0
tensorflow-estimator 2.6.0
tensorflow-gpu 2.6.0
tensorflow-io-gcs-filesystem 0.31.0
keras 2.6.0
cudatoolkit 11.3.1
cudnn 8.2.1
安装步骤
pip install tensorflow-gpu==2.6.0 -i https://pypi.douban.com/simple/
pip install numpy==1.19.3
conda install cudnn==8.2.1
pip install six==1.15.0
pip install typing-extensions==3.7.4
pip install tensorflow-estimator==2.6.0
pip install tensorflow==2.6.0
pip install ipykernel==6.14