tensorflow pytorch CUDA 安装 Win10下安装CUDA11.0+CUDNN8.0+tensorflow-gpu2.4.1+pytorch1.7.0+paddlepaddle-g

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\bin;
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\libnvvp;

# CUDA的安装目录
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0

nvcc -V

pip install -i http://pypi.tuna.tsinghua.edu.cn/simple tensorflow-gpu==2.4.1

import tensorflow as tf
import os
os.environ['TF_CPP_MIN_LOG_LEVEL']='2'
print(tf.__version__)
a = tf.constant(1.)
b = tf.constant(2.)
print(a+b)
print('GPU:', tf.test.is_gpu_available())

pip install torch===1.7.0+cu110 torchvision===0.8.1+cu110 torchaudio===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html

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

python -m pip install paddlepaddle-gpu==2.0.0.post110 -f http://paddlepaddle.org.cn/whl/stable.html

import paddle
paddle.utils.run_check()