当前位置: 代码迷 >> 综合 >> win10_x64 - tensorflow_gpu cuda9.1+cudnn7.1 - anaconda环境搭建
  详细解决方案

win10_x64 - tensorflow_gpu cuda9.1+cudnn7.1 - anaconda环境搭建

热度:54   发布时间:2024-01-04 08:46:01.0

0.保证电脑处于联网状态,笔者电脑配置win10_x64 + gtx 1050 ti,如何查看自己的显卡对应的cuda版本在网上找,此处选择的是cuda9.1+cudnn7.1,如何选择匹配cuda和cudnn的版本在nvidia的网站可以看到,安装文件可以在nividia的网站下载或者其他方法下载。

1.安装包准备

1.1 百度搜索,下载Anaconda3-4.2.0-Windows-x86_64.exe(默认python3.5.2)

1.2 https://github.com/fo40225/tensorflow-windows-wheel找到适合自己cuda和cudnn版本的tensorflow安装包,此处选择的是

tensorflow_gpu-1.5.0-cp36-cp36m-win_amd64.whl 或者使用pip install tensorflow-gpu==1.4.0从网上下载,可以指定版本号,另笔者尝试下,发现也很快。

2.命令行安装

2.1 管理员身份运行: Anaconda Prompt

2.2 查看anaconda的版本: anaconda --version

2.3 python版本选择,查看可用的python版本: conda search --full-name python

2.4 安装指定版本的python(anaconda可以安装多个python环境):

conda env remove -n py36

conda create -n py36 python=3.6

activate py36

deactivate

2.5 安装tesorflow需要的环境:

conda env remove -n tensorflow-gpu

conda create --name tensorflow-gpu python=3.6

2.6 找到下载好的tensorflowxxxx.whl的路径,执行下面的命令:

tensorflow版本不对出错

python -m pip install --upgrade pip

pip install D:/tensorflow_gpu-1.5.0-cp36-cp36m-win_amd64.whl

2.7 验证安装是否成功

python

import tensorflow as tf

hello = tf.constant("hello, Tensorflow!")

sess = tf.Session()

print(sess.run(hello))

致此,安装成功!!!

  相关解决方案