tensorflow
查看NVIDIA基本信息
命令:nvidia-smi
确定cuda版本,可向下兼容,查看目前已安装:
命令:nvcc -V
进入tensorflow官网查看安装
Build from source on Windows | TensorFlow (google.cn)
利用anaconda安装
可以看到,tensorflow2.9对应的CUDA版本是11.2,cuDNN版本是8.1
先安装cudatookit
conda install -c conda-forge cudatoolkit=11.2
后安装cudnn
conda install -c conda-forge cudnn=8.1.0
最后安装tensorflow
pip install tensorflow-gpu==2.9.0
验证是否安装成功
在环境中利用下面几条指令验证:
python
import tensorflow as tf
tf.__version__
tf.test.is_gpu_available()
安装ipykernel
方便在jupyter中使用
pip install ipykernel