下载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
官网下载速度慢,已上传,可以直接下
这里我下载的版本是
![在这里插入图片描述](https://img-blog.csdnimg.cn/20200831173716188.jpg#pic_center
先下载
链接:https://pan.baidu.com/s/1mkhIJC8HeXJ3bT29N1bpsQ
提取码:de7y
参照其他人的文章一步一步下一步基本就可以了,不过值得一提的是安装路径最好默认,这样就直接按照我的教程,如果要安装在其他盘可能要自己注意下我代码路径和你的路径。
https://blog.csdn.net/qq_37296487/article/details/83028394
请看这个
https://blog.csdn.net/u010618587/article/details/82940528
cudnn文件如何放在cuda
https://blog.csdn.net/u011473714/article/details/95042856
这里环境变量我直接写出来你可以复制,我是默认路径
![在这里插入图片描述](https://img-blog.csdnimg.cn/2020083117411064.jpg?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L1E2MzI2NTU2NzI=,size_16,color_FFFFFF,t_70#pic_center
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\lib\x64
%CUDA_PATH%\bin
%CUDA_PATH%\lib\x64
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2
%CUDA_SDK_PATH%\bin\win64
%CUDA_SDK_PATH%\common\lib\x64
C:\ProgramData\NVIDIA Corporation\CUDA Samples\v10.2
![在这里插入图片描述](https://img-blog.csdnimg.cn/20200831174642850.jpg?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L1E2MzI2NTU2NzI=,size_16,color_FFFFFF,t_70#pic_center
这是我安装成功过后的path里变量,仅供参考,
win+R按键打开cmd,输入以下代码安装,强调的是需要在install后加 --user,不然要失败
pip install --user torch=1.6.0 torchvision=0.7.0 -f https://download.pytorch.org/whl/torch_stable.html
版本兼容问题请看这个
https://blog.csdn.net/qq_27825451/article/details/89082978
安装前打一些补丁
python -m pip install --upgrade pip
pip install --user -U --ignore-installed wrapt enum34 simplejson netaddr
补丁好之后安装gpu版本的TensorFlow
pip install tensorflow-gpu=1.15.0(先安装和配置cuda+cudnn)
打开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