windows下安装Tensorflow-gpu-2.5.0
1、安装Anaconda3虚拟环境
官网下载过慢,因此使用清华大学的镜像安装
网址:
https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/
选择64位版本
注意:钩选第二项:
安装完成后开始菜单会出现这么几个应用
2、安装TensorFlow
在安装之前,要保证电脑内安装最新版的显卡驱动、还有与之相符的CUDA与cuDNN,Visual Stdio,可参考之前的博文
(1)、打开Anaconda Prompt,创建新环境
conda create --name tf2 python=3.7 # “tf2”是你建立的conda虚拟环境的名字
conda activate tf2 # 进入名为“tf2”的conda虚拟环境
(2)、然后安装TensorFlow-gpu
安装之前要进行CUDA和CuDnn的安装,参考之前博客
pip install tensorflow-gpu -i https://pypi.douban.com/simple
等待安装完成
(3)、验证
在安装了tensorflow的环境中,输入
python -c "import tensorflow as tf;print(tf.reduce_sum(tf.random.normal([1000, 1000])))"
输出:
`
输出:
安装成功