Windows10+GPU+Anaconda3+Pytorch+TensorFlow+Keras配置方法


补充:测试Tensorflow是否支持GPU的代码。

import tensorflow as tf
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))


配环境这个事真的是常看常新,随着package的不断更新,之前配环境的经验都不能用了,所以新开一贴,记录一下最近配环境的曲折之路。

之前博客Windows10 GPU版Tensorflow配置教程+Anaconda3+Jupyter Notebook依然可以用。

#Windows10+GPU+Anaconda3+Pytorch+TensorFlow+Keras配置方法

最近由于人脸检测,接触到了pytorch框架,首先按照深度学习人脸识别系统 DFace移植WIN64环境详解(支持GPU)配置了一下。但是这位博主的经验也有点久远了,我按照他的来做并没有成功。摸索出来的成功配置步骤可以总结如下:

####1.环境信息
操作系统:Windows10
Anaconda3版本:4.2.0
CUDA版本:8.0
cuDNN版本:5.1
假设这些都已经配置好了,没有的话可以参考
Windows10 GPU版Tensorflow配置教程+Anaconda3+Jupyter Notebook

####2.准备配置文件
(1)environment-win64.yml
这是用来创建Anaconda环境的配置文件
用Notepad++打开,文件第一行可以修改自己想要的environment名称ÿ

  • 1
    点赞
  • 9
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值