Windows下使用Anaconda配置tensorflow-GPU开发环境

#1,下载安装Anaconda
#2,下载安装Nvidia显卡驱动,CUDA tookit以及cuDNN,各自安装解压并确保添加至系统路径
#3,在Anaconda Prompt中创建新的虚拟环境tensorflow-gpu(可自定义命名)
#4,安装numpy以及matplot库
#5,在创建的tensorflow-gpu环境中安装tensorflow-gpu

pip install --ignore-installed --upgrade tensorflow-gpu

#6,测试tensorflow-gpu

from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())

#7,配置tensorflow-gpu到jupyter notebook环境
在tensorflow-gpu环境下

conda intsall anaconda
python -m ipykernel install --user --name tensorflow-gpu --display-name "Python (tensorflow-gpu)"

#8,进入jupyter notebook新建文件测试即可

你可能感兴趣的:(机器学习)