1.安装tensorflow

使用conda创建一个虚拟环境

！！注意！！

pip install -i https://pypi.doubanio.com/simple tensorflow=2.1.0

2.安装cuda10.1

下载地址：

https://developer.nvidia.com/cuda-toolkit-archive

一路next安装完成之后

查看环境变量，默认cuda的环境变量会自动添加

配置path中的环境变量

在终端中测试：

3.cudnn7.6配置文件

文件下载：https://developer.nvidia.com/rdp/cudnn-archive

下载这个文件需要登录NVIDIA官网，我登录了半天登不上去，后面把网页的语言设置成English登上去了

把压缩包解压出来

测试能不能使用GPU

import tensorflow as tf import os os.environ['TF_CPP_MIN_LOG_LEVEL']='2' a = tf.constant(1.) b = tf.constant(2.) print(a+b) print('GPU:', tf.test.is_gpu_available())

GPU环境成功会打印如下

tensorflow-gpu_tensorflow GPU环境安装踩坑日记相关推荐

Linux安装jdk导致linux命令失效(配置JDK环境)（踩坑日记）
在配置jdk环境变量时如果在 vim /etc/profile 进行时一定要自习检查如果有问题你的服务器命令就全部失效了此时你需要打开这个文件从新配置一下,(如果打不开Vim \etc\profil ...
mac es安装踩坑日记
安装es 一开始通过brew安装:brew install elasticsearch 提示报错: elasticsearch has been disabled because it is swit ...
Win 10 + Ubuntu 18.04 双系统安装与深度学习环境配置安装踩坑实录（上篇）
Win 10 + Ubuntu 18.04 双系统安装与深度学习环境配置安装踩坑实录(上篇) 折腾了两三天总算顺利在电脑上完成了装x的双系统安装,一路走来还比较顺利,主要在ubuntu的显卡设置上躺了 ...
Windows环境下PyTorch_geometric安装踩坑
Windows环境下PyTorch_geometric安装踩坑 pytorch geometric在windows环境下安装非常恶心,莫名其妙各种报错.本帖针对GCC的编译error提供解决方案. 一 ...
Win10 TensorFlow（gpu）安装详解
Win10 TensorFlow(gpu)安装详解写在前面:TensorFlow是谷歌基于DistBelief进行研发的第二代人工智能学习系统,其命名来源于本身的运行原理.Tensor(张量)意味着 ...
anaconda+cuda+cudnn+pytorch安装踩坑大全
windows环境下anaconda+pycharm+cuda+cudnn+pytorch安装踩坑大全 anaconda安装 pycharm安装 cuda cudnn pytorch gym环境安装 ...
全志哪吒D1-H Tina Linux Ubuntu 22.04入门踩坑日记
哪吒D1-H Tina Linux入门踩坑日记系统环境源码编译 mklibs-readelf的C++标准问题 m4的SIGSTKSZ问题 libfakeroot的_STAT_VER问题 read_ ...
nnUNet安装踩坑记录
nnUNet安装踩坑记录论文地址:https://arxiv.org/abs/1809.10486 https://github.com/MIC-DKFZ/nnUNet 更新于2020/12/04: ...
Android Studio安装踩坑
title: Android Studio安装踩坑 date: 2018-09-07 19:31:32 updated: tags: [Android,Android Studio,坑] descri ...

tensorflow-gpu_tensorflow GPU环境安装踩坑日记

1.安装tensorflow

2.安装cuda10.1

3.cudnn7.6配置文件

tensorflow-gpu_tensorflow GPU环境安装踩坑日记相关推荐

最新文章

热门文章