下载TensorFlow对应版本请看这个网站
https://blog.csdn.net/qq_27825451/article/details/89082978
下载pytorch请看官网
https://pytorch.org/
CUDA:https://docs.nvidia.com/cuda/cuda-installation-guide-microsoft-windows/index.html
cuDNN:https://docs.nvidia.com/deeplearning/sdk/cudnn-install/index.html#installwindows
官网下载速度慢,已上传,可以直接下
这里我下载的版本是

打开cmd或者anaconda的
进入目录
cd C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\bin
nvcc -V
cd C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\extras\demo_suite
输入执行
bandwidthTest.exe
deviceQuery.exe
结果出现
Result = PASS
就代表成功了
import tensorflow as tf
import torch