前言
台式机换了个RTX-3080的显卡,要重新装机并配置tensorflow的环境,踩了超多的坑,现在记录一下成功的经历,真的太不容易了…我系统用的Windows,网上大多数人用的都是Linux系统,但是我不想再花时间去学一个系统了。网上说30系列显卡只支持cuda11及以上,这里也不尝试cuda10了。而且我之前自己搭网络用的tf2.0,本文也不尝试安装tf1.0的版本了。先列出我最后成功的版本:
- 系统:Windows10
- 显卡:RTX-3080
- 驱动:456.43
- cuda:CUDA11.1.0_win10_network
- cudnn: cudnn-11.1-windows-x64-v8.0.5.39
- tensorflow:tf-nightly-gpu 2.5.0.dev20201217
一、安装Anaconda
打开清华镜像站:https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/?C=M&O=D
我下了个最新的版本:Anaconda3-2020.11-Windows-x86_64.exe
安装过程,我这次做出的改动有:
1.安装Install for:“All Users”
2.直接自动添加环境变量了,即勾选了“Add Anaconda3 to the system PATH environment variable”
二、查看驱动
RTX3080采用了新的Ampere架构GA102-200,那么显卡驱动也必然是需要最新的,可以参考cuda与驱动对应的关系:
想要装CUDA 11.1的话,驱动版本要在456.38以上。想要看自己的驱动版本的话,可以在cmd中输入:
nvidia-smi
可以看到驱动版本是456.55,还显示了CUDA版本为11.1。
也可以不提前安装驱动,CUDA11安装里面自带了驱动,因为我后面卸载重装了,就直接用的CUDA11安装的驱动(自动安装的版本为456.43),如下所示:
三、安装CUDA
-
到链接: 英伟达官网下载对应的CUDA版本。
我下载的版本是CUDA11.1.0 ,选择"Windows;x86_64;10;exe(network)" -
安装CUDA:双击执行下载的exe文件,会先解压文件到临时目录(不是安装目录),保持默认即可.【我不知道可不可以下到别的里面,就照着博客下了】
- 这里需要选择自定义安装:
- 自定义安装选项,不勾选“Visual Studio Integration”,勾选的话后面会有个什么提示,所以又退回这步不勾选了。
- 这三个地址十分重要,后续我们将进行环境变量的配置,不要改变路径。
- 之后等待安装,遇到了下面的情况
Installed:
- Nsight Monitor
Not Installed:
- Nsight for Visual Studio 2019
Reason: VS2019 was not found
- Nsight for Visual Studio 2017
Reason: VS2017 was not found
- Nsight for Visual Studio 2015
Reason: VS2015 was not found
- Integrated Graphics Frame Debugger and Profiler
Reason: see https://developer.nvidia.com/nsight-vstools
- Integrated CUDA Profilers
Reason: see https://developer.nvidia.com/nsight-vstools
不知道有没有什么问题,先继续往下吧。
- 配置CUDA环境变量
网上很多说要环境变量>系统变量>Path中手动添加路径的,我整合了好多条:
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\bin【自动配的】
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\libnvvp【自动配的】
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\lib
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\lib\x64
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\include
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\extras\CUPTI\lib64
C:\ProgramData\NVIDIA Corporation\CUDA Samples\v11.1\bin\win64
C:\ProgramData\NVIDIA Corporation\CUDA Samples\v11.1\common\lib\x64
但最后发现,其实不用手动配置,用自动配的前两条也可以运行程序啊…不知道为什么那么多人都说要配置路径…
- 我们可以验证一下cuda是否成功,按win+R 输入cmd,输入命令:
nvcc -V
得到如下的结果: