windows配置cudatoolkit10.1、cudnn7.6和tensorflow2.1

安装之前要注意以下自己的硬件版本,而且tensorflow版本、cudatoolkit、cudnn版本是有对应的

1、下载anacona的python3.7版本,这里下载使用北外的镜像下载,速度巨快!!!
https://mirrors.bfsu.edu.cn/anaconda/archive/

 2、换源,让你随后的安装飞起来

打开Anaconda Prompt

 

 键入

conda config --set show_channel_urls yes

在C:\Users\usrname中找到.condarc文件

把里面的内容改为

channels:
  - defaults
show_channel_urls: true
default_channels:
  - https://mirrors.bfsu.edu.cn/anaconda/pkgs/main
  - https://mirrors.bfsu.edu.cn/anaconda/pkgs/r
  - https://mirrors.bfsu.edu.cn/anaconda/pkgs/msys2
custom_channels:
  conda-forge: https://mirrors.bfsu.edu.cn/anaconda/cloud
  msys2: https://mirrors.bfsu.edu.cn/anaconda/cloud
  bioconda: https://mirrors.bfsu.edu.cn/anaconda/cloud
  menpo: https://mirrors.bfsu.edu.cn/anaconda/cloud
  pytorch: https://mirrors.bfsu.edu.cn/anaconda/cloud
  simpleitk: https://mirrors.bfsu.edu.cn/anaconda/cloud

 

回到Anaconda Prompt

conda create -n TF2.1 python=3.7#新建一个名叫TF2.1的环境
conda activate TF2.1#激活环境
conda install cudatoolkit=10.1#安装cudatoolkit
conda install cudnn=7.6

在使用豆瓣源安装trensorflow

python -m pip install --upgrade pip#安装之前先更新一下pip
pip install tensorflow==2.1 -i https://pypi.doubanio.com/simple/
#在python里面测试一下
(TF2.1) C:\Users\daoliang>python Python 3.7.0 (default, Jun 28 2018, 08:04:48) [MSC v.1912 64 bit (AMD64)] :: Anaconda, Inc. on win32 Type "help", "copyright", "credits" or "license" for more information. >>> import tensorflow as tf 2021-01-27 20:02:13.831711: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cudart64_101.dll >>> tf.__version__ '2.1.0' >>> ^Z (TF2.1) C:\Users\daoliang>

 

posted @ 2021-01-27 20:07  潜心_修炼  阅读(490)  评论(0编辑  收藏  举报