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>