通過(guò) nvidia-smi 查看到的 CUDA version 為系統(tǒng)最大可裝的 CUDA的版本
安裝tensorflow-gpu
首先確定tensorflow和python的版本（python3.6支持多數(shù)版本扭勉，是個(gè)不錯(cuò)的選擇）
查看tensorflow對(duì)應(yīng)的tensorflow-gpu所需要的cuda和cudnn版本
https://www.tensorflow.org/install/source#gpu
創(chuàng)建環(huán)境 tf_gpu
- conda create -n tf_gpu python=3.6
- conda install cudnn=7.6.5
- conda install cudatoolkit=10.1
- conda install tensorflow-gpu=2.1
python 檢測(cè)
import tensorflow as tf
print(tf.__version__)
print(tf.test.is_gpu_available())
print(tf.config.list_physical_devices('GPU'))
