Win10系统下Tensoflow的版本搭配
机器学习中第一步就是环境的配置,最难顶的就在于版本的搭配。本次搭配是在Win10
笔记本上进行的,记录一下以便分享。
Cuda
版本:9.0
Download_Cuda
旧版本的Cuda
可以在Archive of Previous CUDA Releases
里找到:
查看Cuda
相关的编译器支持:
显示Cuda9.0
支持Visual Studio 2017
:
Cudnn
版本:7.1
注册一个Nvidia
账号按照提示操作即可:
Download_Cudnn
Cudnn
安装:将下载好的安装包解压后放至Cuda相对应的文件夹下。
这里不能直接
将文件夹
拖至Cuda文件夹下覆盖粘贴,一定是将每个文件夹下的内容
复制到Cuda中相应文件夹下
。否则Cuda
会损坏。
使用Anaconda
创建虚拟环境:
Download_Anaconda
# 创建虚拟环境
conda create -n env_name python=3.6
# 激活虚拟环境
conda activate env_name
# 退出虚拟环境
conda deactivate
Win10
命令提示符中直接使用Conda
指令:
在系统变量中添加anaconda
目录下Scripts
文件夹的路径:
使用pip
指令在虚拟环境中安装tensorflow
:
# 进入虚拟环境
conda activate env_name
# 查看当前安装的包
pip list
# 安装tensorflow,这里若电脑里有GPU,则安装tensorflow-gpu
pip install tensorflow-gpu==1.12