0
点赞
收藏
分享

微信扫一扫

TensorFlow-GPU安装链接整理

骨灰级搬砖工 2022-04-25 阅读 58
tensorflow

在安装TensorFlow-GPU时需要对应TensorFlow-GPU版本与CUDA、CUDNN的版本对应。每次需要更改环境的时候,都要各种找网址。这里做一个网址整理:

1、安装GPU版本之前先检查电脑是否支持GPU版本

链接: https://pan.baidu.com/s/19Sf1WUizsMDhGm4ax86hWA?pwd=df12icon-default.png?t=M3K6https://pan.baidu.com/s/19Sf1WUizsMDhGm4ax86hWA?pwd=df12 提取码: df12

双击exe,安装软件选择否。

 1代表的是显卡驱动版本,2代表的是支持CUDA计算

2、查找要安装的gpu版本和对应cuda、cudnn之间的版本对应关系

https://www.tensorflow.org/install/source_windows3、查看电脑显卡驱动版本是否符合对应CUDA版本要求

除了前面使用TechPowerUp GPU-Z 软件查看外还可以使用:       cmd输入nvidia-smi

nvidia-smi显示不是内部或外部命令也不是可运行的程序

 链接:Release Notes :: CUDA Toolkit Documentation

4、下载对应版本的CUDA 

CUDA Toolkit Archive | NVIDIA Developer

 题外话,下载太慢可以安装一个IDM

链接: https://pan.baidu.com/s/1pFEYiwEMFHvyNPjttpDH_g?pwd=m35d icon-default.png?t=M3K6https://pan.baidu.com/s/1pFEYiwEMFHvyNPjttpDH_g?pwd=m35d 提取码: m35d 

5、下载对应版本的CUDNN

cuDNN Archive | NVIDIA Developer(需要注册账号)

6、安装CUDA和CUDNN

 Cuda和cuDNN安装教程(超级详细)_jhsignal的博客-CSDN博客_cudnn安装

举报

相关推荐

0 条评论