死磕装环境——tensorflow_gpu 2.0
谨以此文献给my cute baby lxx
原视频
首先打开tensorflow官网查看需要安装的东西
NVIDIA GPU drivers
查看自己显卡类型,在官网下载驱动安装,安装时安装图形驱动
CUDA Toolkit
选择10.0版本安装就好了
安装后c盘会有如下目录C:\Program Files\NVIDIA GPU Computing Toolkit
CUPTI(这个不用管)
cuDNN SDK
这个东西要登录NVIDIA官网才能下载,但是家里的网络登不上,后来是在网盘下载的
TensorRT
这个也是要登录才能下载,但不是必选项,所以我放弃了
安装驱动和CUDA并下载cuDHH后,Tensorflow官网对上述文件有路径要求
把cuDNN解压出来的cuda文件夹放在c盘tools目录下(如果没有就创建一个)
之后按上图依次添加环境变量
然后在conda命令行中输入(anaconda或者miniconda都行)
conda env create -v -f tensorflow-gpu.yml
这个yml文件在github上
然后就是把这个环境加入jupyter botebook内核
python -m ipykernel install --user --name tensorflow --display-name “Python 3.7 (tensorflow)”
安装完成后可以在Anaconda Navigator中看到此环境,有162个包
在jupyter notebook中测试: