TensorFlow环境安装CUDA_cuDNN_Py37_TF-GPU/Py37_TF-CPU/Py27

前言

在执行下面的虚拟环境创建之前,需要先安装Anaconda(最新版本即可)。若创建过程中包的下载速度较慢,可以考虑换源(方法网上搜索)。

所用到的所有文件:
链接:https://pan.baidu.com/s/1J_YWJXJPpTh-DgKFAlzeCA
提取码:afrx

py37tfgpu

Python3.7.13+TensorFlow-gpu2.3.0+cuDNN7.6.5+CUDA10.1

安装CUDA10.1

  1. 打开cuda_10.1.168_425.25_win10.exe

  2. 自定义安装,只选择CUDA单元,并且取消其下的VS Integration

    image-20220411081546151

  3. 检查一下系统变量

    image-20220410154403002

    image-20220411090707367

  4. Ctrl + R,输入cmd进入命令行,输入nvcc -V查看cuda版本号,若出现则安装成功

    image-20220410154626298

安装cuDNN7.6.5

  1. 解压cudnn-10.1-windows10-x64-v7.6.5.32.zip
  2. 将其中的所有文件复制到路径C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1下

安装Py3.7.13+TF2.3.0

在Anaconda Prompt中执行 注意顺序

1. 创建环境(删除环境为conda remove -n [env_name] --all,然后在anaconda\envs文件夹下删除[env_name]文件夹)
conda create -n py37tfgpu python=3.7.13
2. 进入环境(退出环境用conda deactivate)
conda activate py37tfgpu
conda install cudatoolkit=10.1
conda install cudnn=7.6.5
3. 安装tensorflow-gpu
conda install tensorflow-gpu==2.3.0

测试

在Anaconda Prompt中执行

1. 输入python进入代码编辑
2. 输入以下程序
import tensorflow as tf
print(tf.test.is_built_with_cuda())
print(tf.test.is_gpu_available())

输出True True,则安装成功。

py37tfcpu

Python3.7.13+TensorFlow2.3.0

1. 创建环境(删除环境为conda remove -n [env_name] --all,然后在anaconda\envs文件夹下删除[env_name]文件夹)
conda create -n py37tfcpu python=3.7.13
2. 进入环境(退出环境用conda deactivate)
conda activate py37tfcpu
3. 安装tensorflow
conda install tensorflow==2.3.0

测试

在Anaconda Prompt中执行

1. 输入python进入代码编辑
2. 输入以下程序
import tensorflow as tf
print(tf.__version__)
a = tf.constant(1.)
b = tf.constant(2.)
print(a+b)

输出2.3.0 tf.Tensor(3.0, shape=(), dtype=float32),则安装成功。

py27

1. 创建环境(删除环境为conda remove -n [env_name] --all,然后在anaconda\envs文件夹下删除[env_name]文件夹)
conda create -n py27 python=2.7
2. 进入环境(退出环境用conda deactivate)
conda activate py27
3. 测试
1)输入python进入代码编辑
2)输入以下程序
print 'Hello World!'
3)输出Hello World安装成功

由于2020年之后Python2.7停止维护,导致tensorflow 1安装失败,到还是在百度网盘中放置了可能用到的文件

image-20220411120119315

猜你喜欢

转载自blog.csdn.net/qq_42283621/article/details/124095937
py