TensorFlow 2.x GPU版在conda虚拟环境下安装步骤

先下载安装驱动：https://www.nvidia.cn/Download/index.aspx?lang=cn，版本要求
WSL cuda 驱动 https://developer.nvidia.com/cuda/wsl
下载安装 anaconda，管理虚拟环境：https://www.anaconda.com/products/individual，并换国内的源，加速后面下载包
linux vim ~/.bashrc 添加 export PATH=/usr/local/bin:$PATH（conda路径自己改），source ~/.bashrc
打开 conda 命令行，conda create -n env_name python=3.7，or 自定义路径 conda create --prefix=D:\yourpath\tf2 python=3.7
激活创建的虚拟环境，activate env_name or activate D:\yourpath\tf2
去查需要安装的 tf 版本对应的 cuda 和 cudnn 的版本（以下是 tf 2.3 的 gpu版）
conda install cudatoolkit=10.1，conda install cudnn=7.6，pip install tensorflow-gpu==2.3
如果速度很慢，可以 pip install -i https://mirrors.aliyun.com/pypi/simple/ tensorflow-gpu==2.3

-i http://mirrors.aliyun.com/pypi/simple --trusted-host mirrors.aliyun.com

cudatoolkit 历史版本：https://developer.nvidia.com/cuda-toolkit-archive
cudnn 下载慢可以去官网下载，并解压到虚拟环境路径下：

遇到下载很慢的包：

也可以去清华源下载安装包放到： /anaconda3/pkgs
使用离线包安装如：conda install --use-local pytorch-1.9.0-py3.8_cuda11.1_cudnn8.0.5_0.tar.bz2

CUDA 版本要求链接：https://www.tensorflow.org/install/source_windows
安装包下载： https://blog.csdn.net/qq_27825451/article/details/89082978

版本匹配见下图

测试安装正确

import tensorflow as tfversion = tf.__version__
gpu_ok = tf.test.is_gpu_available()
# tf.config.list_physical_devices('GPU') # 也可以
print("tf version:", version, "\nuse GPU", gpu_ok)

输出：

tf version: 2.3.1
use GPU True

如果显示都成功找到GPU了，还报错，可能是你的显卡太旧了，尝试降低 tf 版本，或者换显卡，也有可能，显存不够，你可以调小点 batch_size

conda常用命令

conda list：查看安装了哪些包
conda install package_name：安装包
conda env list 或 conda info -e：查看已安装虚拟环境
conda update conda：更新conda
conda remove --prefix=D:\yourpath\tf2x --all ：删除指定路径环境

TensorFlow 2.x GPU版在conda虚拟环境下安装步骤相关推荐

Python 的scikit-learn和tensorflow在虚拟环境下安装配置笔记
Python 机器学习安装配置笔记这学期开始上人工智能课,任课老师要求安装python和对于机器学习深度学习的各种库. 以前做过类似项目,顺便就把配置环境的流程记录下来. 中间有引用他人的博客,都有 ...
conda虚拟环境下使用pip安装包报错Retrying (Retry(total=0, connect=None, read=None, redirect=None, status=None)
Q1: conda虚拟环境下使用pip安装软件报错: WARNING: Retrying (Retry(total=0, connect=None, read=None, redirect=None, ...
在conda虚拟环境中安装ipython
大家好,我是爱编程的喵喵.双985硕士毕业,现担任全栈工程师一职,热衷于将数据思维应用到工作与生活中.从事机器学习以及相关的前后端开发工作.曾在阿里云.科大讯飞.CCF等比赛获得多次Top名次.现 ...
Mac-OSX的Python3.5虚拟环境下安装Opencv
Mac-OSX的Python3.5虚拟环境下安装Opencv 1 关键词关键词:Mac,OSX,Python3.5,Virtualenv,Opencv 2 概述本文是一篇环境搭建的基础 ...
从零开始学习opencv——在虚拟环境下安装opencv环境
毕设准备做cv相关项目,今天开始学习cv基础知识,课程为B站"[不要再看那些过时的OpenCV老教程了]2022巨献,OpenCV零基础小白最新版全套教程(人工智能机器视觉教程)" ...
【TensorFlow】Windows10 64位下安装TensorFlow - 官方原生GPU版安装
之前写过一篇在 ubuntu 下安装 TensorFlow 的教程,那个时候 TensorFlow 官方还不支持 Windows 系统,虽然可以通过其他方法安装,但是终究不是原生的,而且安装过程繁琐 ...
conda虚拟环境中安装ipython
问题:今天安装了tensorflow,启动ipython竟然提示如下: In [1]: import tensorflow as tf -------------------------------- ...
conda虚拟环境中安装 jupyter notebook
@创建于:2022.01.27 @修改于:2022.01.27 文章目录需要在conda的某个特定虚拟环境中安装Jupyter Notebook. # 进入虚拟环境(windows系统) conda ...
conda虚拟环境内安装CUDA9.0
1.下载CUDA9.0 下载地址如果ubuntu版本过高,则需在conda环境内安装指定版本的gcc,可以参考这里. 选择runfile下载,下载完成之后执行如下语句,运行.run文件 sh cud ...

TensorFlow 2.x GPU版在conda虚拟环境下安装步骤

TensorFlow 2.x GPU版在conda虚拟环境下安装步骤相关推荐

最新文章

热门文章