1.版本说明
- Anaconda3 == 4.8.3
下载地址: https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/Anaconda3-2020.07-Linux-x86_64.sh - Tensorflow-gpu==2.1.0
我的机器有GPU,所以选择gpu版本的。 - NVIDIA驱动==418.165.02
这个NVIDIA驱动版本选择根据选择的tensorflow版本和cuda对应关系确定。我打算用anaconda安装tensorflow-gpu2.1.0所有我的cuda版本是10.1。使用Anaconda的好处是不用自己装cuda和cuDNN,这样可以避免很多版本不兼容的错误。
tensorflow和cuda版本对应
cuda各个版本的nvidia驱动版本要求
(1)先查看显卡类型
nvidia-smi -L
#==========================================================
GPU 0: Tesla V100-PCIE-32GB (UUID: GPU-**********)
GPU 1: Tesla V100-PCIE-32GB (UUID: GPU-**********)
GPU 2: Tesla V100-PCIE-32GB (UUID: GPU-**********)
#===========================================================
(2)在