最详细 win10 tensorflow2.0 和 pytorch1.20 gpu 版安装全程指导

官网安装说明
首先该链接非常重要,即使以后tensorflow更新了,我的博客不再适用了该链接依然可以助你安装成功

1.首先检查你的gpu

tensorflow只支持nvidia的显卡,AMD的同学很不幸下面没你事了
根据官网建议你的显卡的算力至少应该在3.5以上,
你可以在nvidia官网查看你的 gpu 的 compute capability

2.安装gpu driver

https://www.nvidia.com/download/index.aspx?lang=en-us
填写你的gpu信息后点search,如果没有搜索结果 do this:
在这里插入图片描述
然后安装:
在这里插入图片描述
在这里插入图片描述
安装完成:
在这里插入图片描述

3.安装CUDA

注意参考官网安装说明安装合适的版本
CUDA下载地址

在这里插入图片描述

4.安装CUDNN

安装与CUDA相匹配的CUDNN版本:
cudnn下载地址
注意点 I agree后才会出现各个版本,并且cudnn需要登陆才能下载。
在这里插入图片描述

5.配置CUDA与CUDNN的环境变量

环境配置参见官网安装说明

在这里插入图片描述

6.安装TensorRT(可选)

注意不要下最新版!!!
注意不要下最新版!!!
注意不要下最新版!!!
谷歌跟不上!
TensorRT5.0下载链接

7.安装 Miniconda 而非Anconda(它比Anconda小很多,我比较喜欢)

Miniconda下载链接
8.miniconda安装tensorflow-gpu=2.0
直接安装就行

9.测试

在这里插入图片描述
注意上面图片的最后一句花了大概好几分钟,后面的就快了,第一次运行可能需要加载一些东西。
用jupyter测试

在这里插入图片描述
官方案例演示

10.然后安装pytorch:

conda install pytorch torchvision cudatoolkit=10.0 -c pytorch
Downloading and Extracting Packages
pytorch-1.2.0        | 479.3 MB  |                                                                                                                                                   |   0%
torchvision-0.4.0    | 2.3 MB    |                                                                                                                                                   |   0%

CondaHTTPError: HTTP 000 CONNECTION FAILED for url <https://conda.anaconda.org/pytorch/win-64/pytorch-1.2.0-py3.7_cuda100_cudnn7_1.tar.bz2>
Elapsed: -

An HTTP error occurred when trying to retrieve this URL.
HTTP errors are often intermittent, and a simple retry will get you on your way.

CondaHTTPError: HTTP 000 CONNECTION FAILED for url <https://conda.anaconda.org/pytorch/win-64/torchvision-0.4.0-py37_cu100.tar.bz2>
Elapsed: -

An HTTP error occurred when trying to retrieve this URL.
HTTP errors are often intermittent, and a simple retry will get you on your way.

可见其他依赖在清华源上都有,但pytorch-1.2.0-py3.7_cuda100_cudnn7_1.tar.bz2和torchvision-0.4.0-py37_cu100.tar.bz2没有,那就单独下载后安装:
在这里插入图片描述
在这里插入图片描述
验证是否安装成功:
在这里插入图片描述

发布了588 篇原创文章 · 获赞 170 · 访问量 29万+

猜你喜欢

转载自blog.csdn.net/qq_41228218/article/details/105210377