本机配置:
Ubuntu 20.04.2 LTS
nvidia-smi 460.39
cuda 11.0.2
cudnn 8.0.4.30
pytorch 1.7.1
cudatoolkit-11.0.221
anaconda env python 3.7 下
pip3 install --upgrade tensorflow-gpu
tensorflow 2.4.1
本机配置:
Ubuntu 20.04.2 LTS
nvidia-smi 460.39
cuda 11.0.2
cudnn 8.0.4.30
pytorch 1.7.1
cudatoolkit-11.0.221
anaconda env python 3.7 下
pip3 install --upgrade tensorflow-gpu
tensorflow 2.4.1