深度学习各种环境问题积累

1. Pytorch

首先要安装anaconda:

推荐清华镜像
https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/
拖到最后下载最新版即可。

这里写图片描述

安装完毕要安装pytorch运行环境:

# If your main Python version is not 3.5 or 3.6
conda create -n test python=3.6 numpy pyyaml mkl

然后是安装gpu版本pytorch:

# for Windows 10 and Windows Server 2016, CUDA 8
conda install -c peterjc123 pytorch cuda80

因为TensorFlow只能用 CUDA8所以我安装的是 cuda8你也可已安装新版本试试

这里写图片描述

2. Keras 

  2.1 Graphviz 安装

os.environ["PATH"] += os.pathsep + 'D:/Program Files (x86)/Graphviz2.38/bin/'
posted @ 2018-12-12 11:05  小小川流  阅读(220)  评论(0编辑  收藏  举报