cuda11.0 + cudnn8.0.5 + anaconda + pytorch + tensorflow

1、conda安装：https://blog.csdn.net/qq_41101213/article/details/115469136

2、cuda 11.0安装
3、cudnn 8.0.5解压安装

一、验证cuda 安装成功

命令1：nvcc -V

命令2：nvidia-smi

二、torchversion==1.7.1

pip install torch==1.8.0 torchvision==0.9.0 -i https://pypi.tuna.tsinghua.edu.cn/simple

传送门

命令：conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch -c conda-forge

torch                  1.7.1
torchaudio             0.7.2
torchsummary           1.5.1
torchvision            0.8.2 

验证：
python

import torch
print( torch.__version__ )
print( torch.cuda.is_available() )
print( torch.cuda.device_count() )

(DL) C:\Users\Bingooo>pip list
Package                Version
---------------------- -------------------
absl-py                0.12.0
astunparse             1.6.3
autopep8               1.6.0
cachetools             4.2.1
certifi                2020.12.5
chardet                4.0.0
click                  7.1.2
cycler                 0.10.0
DNN-printer            0.0.2
Flask                  1.1.2
flatbuffers            1.12
gast                   0.3.3
google-auth            1.28.0
google-auth-oauthlib   0.4.4
google-pasta           0.2.0
grpcio                 1.32.0
h5py                   2.10.0
idna                   2.10
importlib-metadata     3.10.0
itchat                 1.3.10
itsdangerous           1.1.0
Jinja2                 2.11.3
Keras                  2.4.3
Keras-Preprocessing    1.1.2
kiwisolver             1.3.1
Markdown               3.3.4
MarkupSafe             1.1.1
matplotlib             3.3.4
numpy                  1.19.5
oauthlib               3.1.0
olefile                0.46
opencv-python          4.5.1.48
opt-einsum             3.3.0
Pillow                 8.2.0
pip                    22.0.4
protobuf               3.15.7
pyasn1                 0.4.8
pyasn1-modules         0.2.8
pycodestyle            2.8.0
pyparsing              2.4.7
pypng                  0.0.20
PyQRCode               1.2.1
python-dateutil        2.8.1
PyYAML                 5.4.1
redis                  3.5.3
requests               2.25.1
requests-oauthlib      1.3.0
rsa                    4.7.2
scipy                  1.6.3
setuptools             52.0.0.post20210125
six                    1.15.0
tensorboard            2.4.1
tensorboard-plugin-wit 1.8.0
tensorflow-estimator   2.4.0
tensorflow-gpu         2.4.1
termcolor              1.1.0
torch                  1.7.1
torchaudio             0.7.2
torchsummary           1.5.1
torchvision            0.8.2
tornado                6.1
typing-extensions      3.7.4.3
urllib3                1.26.4
Werkzeug               1.0.1
wheel                  0.36.2
wincertstore           0.2
wrapt                  1.12.1
zipp                   3.4.1

三、tensorflow-gpu==2.4.0

版本参考：https://tensorflow.google.cn/install/source_windows#gpu

命令：
此处安装时需要关闭代理！
pip install tensorflow-gpu==2.4.0

（conda search tensorflow-gpu 安装失败，无2.4.0）

验证：
python
import tensorflow as tf

print(tf.version.VERSION)
print(tf.test.is_built_with_cuda()) 
print(tf.test.is_gpu_available())

四、查看cuda和cudnn

torch.version.cuda
torch.backends.cudnn.version()
torch.cuda.get_device_name()