写在最前
这篇博客写给所有具备一定环境搭建基础的同学。
一些具体的步骤我会省去,但会提醒所有我觉得重要的地方。
在摩拳擦掌自己尝试安装之前,阅读这篇文章可以建立一个比较清晰的脉络。
折腾了两天之后来写这篇博客,有些地方可能出现一两个手滑打错的地方,还请诸位加以鉴别,提出改正。
为了释放3080显卡(甚至是30系显卡)的全部算力,须用最新的
tf-nightly-gpu=2.6.0 (预览版tensorflow)
cuda=11.2
cudnn=8.1.0 (for cuda 11.2)
以及Linux (Ubuntu or Lubuntu >=18.04)
是的没错,我测试了windows上面的各种可能组合,速度都要比linux慢1/3!
windows 安装
1. cuda cudnn安装之前一定注意,在uninstall programme工具里面卸载掉所有和nvidia相关的软件!
cuda安装的时候,可以选择勾选是否安装驱动。这里我想说,显卡驱动版本不是特别重要。所以就用cuda自带的驱动也可。
2. 其实在win平台安装cuda要比Linux简单
因为修改环境变量的时候,可以直接找到
系统属性-高级设置-环境变量
对上下两个框分别进行路径添加(安装cuda的时候一般会自动添加一些,可以酌情手动添加),或修改(如替换不同版本的cuda文件夹)。
主要就是添加cuda相关的路径,如(x.x为对应版本,例