当前位置: 首页 > news >正文

使用Windows Linux 子系统安装 Tensorflow,并使用GPU环境

在Microsoft Store商店安装Ubuntu 20.04

使用 nvidia-smi 命令查看GPU信息,查看支持的CUDA版本,这里最高支持11.7

安装cuda工具集

进入官网:CUDA Toolkit Archive | NVIDIA Developer,现在对应版本,点击

配置平台,根据平台选择,安装方式提供了三种:1、下载deb文件配置本地环境安装;2、通过网络加载deb文件安装;3、通过配置文件安装;

这里选择用本地deb方式,安装官方提供的命令安装:

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pinsudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600wget https://developer.download.nvidia.com/compute/cuda/11.7.0/local_installers/cuda-repo-ubuntu2004-11-7-local_11.7.0-515.43.04-1_amd64.debsudo dpkg -i cuda-repo-ubuntu2004-11-7-local_11.7.0-515.43.04-1_amd64.debsudo cp /var/cuda-repo-ubuntu2004-11-7-local/cuda-*-keyring.gpg /usr/share/keyrings/sudo apt-get updatesudo apt-get -y install cuda

添加环境变量:

vim ~/.bashrc

在末尾添加以下内容:

export PATH=$PATH:/usr/local/cuda-11.7/bin

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.7/lib64

source ~/.bashrc

使用 nvcc -V 命令查看安装版本信息:

安装 cuDNN

上官网下载对应版本:https://developer.nvidia.com/rdp/cudnn-archive,下压缩包

将对应的文件包解压后,将文件拷贝到对音乐目录

tar -xvf cudnn-linux-x86_64-8.9.7.29_cuda11-archive.tar.xz

cd cudnn-linux-x86_64-8.9.7.29_cuda11-archive

sudo cp lib/* /usr/local/cuda-11.7/lib64/

sudo cp include/* /usr/local/cuda-11.7/include/

sudo chmod a+r /usr/local/cuda-11.7/lib64/*

sudo chmod a+r /usr/local/cuda-11.7/include/*

至此gpu NN开发环境就构建好了

安装tensorflow环境

安装Anaconda

官网下载安装文件 Download Now | Anaconda

下载下来是一个Anaconda3-2024.06-1-Linux-x86_64.sh文件,执行脚本文件进行安装:

bash Anaconda3-2024.06-1-Linux-x86_64.sh

根据提示选择安装路径,推荐使用sudo安装到/usr/local/anaconda3,添加环境变量:

cd /usr/local/anaconda3/bin && conda init && cd -

设置默认不启动base环境

conda config --set auto_activate_base false

# 构建环境

conda create --name py_3.7.5_tf_gpu_env python=3.7.5 anaconda tensorflow-gpu

# 运行环境

conda activate py_3.7.5_tf_gpu_env

测试环境:

import tensorflow as tf

tf.test.is_gpu_available()


 

http://www.lryc.cn/news/403790.html

相关文章:

  • C++案例三:猜数字游戏
  • LNMP架构部署及应用
  • 【医学影像】X86+FPGA:支持AI医学影像设备应用的工控主板,赋能CT、MRI、X线、超声等医学影像设备
  • 【PostgreSQL】PostgreSQL简史
  • Linux的热插拔UDEV机制和守护进程
  • laravel框架基础通识-新手
  • c++ extern 关键字
  • Meta KDD Cup 2024 CRAG: Comphrehensive RAG Benchmark参赛指南(写了一半跑去改大论文了所以没正式参赛)
  • 系统架构设计师教程 第3章 信息系统基础知识-3.7 企业资源规划(ERP)-解读
  • windows实现自动化按键
  • 阿里云 https证书部署
  • DX-10A信号继电器 柜内安装,板前接线 约瑟JOSEF
  • 芯片光刻后的晶片多层组合构成的吗
  • OpenGL-ES 学习(7) ---- VBO EBO 和 VAO
  • github如何实现和gitlab的同步
  • 内网隧道——隧道技术基础
  • NAS新品“翻车”后,绿联科技要上市了
  • kafka-client如何打印连接日志
  • 1、springboot3 vue3开发平台-后端-项目构建
  • 修改了mybatis的xml中的sql不重启服务器如何动态加载更新
  • Intel和AMD用户再等等!微软确认Win11 24H2年底前登陆
  • Web开发:图片九宫格与非九宫格动态切换效果(HTML、CSS、JavaScript)
  • 价格较低,功能最强?OpenAI 推出 GPT-4o mini,一个更小、更便宜的人工智能模型
  • 【学习笔记】无人机系统(UAS)的连接、识别和跟踪(八)-无人机探测与避让(DAA)机制
  • 网络结构-组件-AI(九)
  • 弹性网络回归(Elastic Net Regression)
  • 【深度学习】FaceChain-SuDe,免训练,AI换脸
  • Uniapp鸿蒙项目实战
  • 计算机三级嵌入式笔记(一)—— 嵌入式系统概论
  • react Jsx基础概念和本质