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

windows10在wsl上利用GPU运行tensorflow 2.12

从tensorflow 2.10开始,已经没有tensorflow-gpu在windows平台,合并为tensorflow,要想使用gpu只能wsl上运行。
大致方法是:安装WSL,安装miniconda,创建conda环境,安装tensorflow,安装pycharm专业版,调用wsl上的python解释器。

具体命令:

wsl --install
重启
菜单栏打开ubuntu,设置用户名,密码

curl https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -o Miniconda3-latest-Linux-x86_64.sh
bash Miniconda3-latest-Linux-x86_64.sh

conda create --name tf2.12 python=3.9
conda activate tf2.12
conda install -c conda-forge cudatoolkit=11.8.0
pip config set global.extra-index-url https://pypi.mirrors.ustc.edu.cn/simple
pip install nvidia-cudnn-cu11==8.6.0.163

CUDNN_PATH=$(dirname $(python -c "import nvidia.cudnn;print(nvidia.cudnn.__file__)"))
export LD_LIBRARY_PATH=$CONDA_PREFIX/lib/:$CUDNN_PATH/lib:$LD_LIBRARY_PATH

pip install tensorflow==2.12
conda install -c nvidia cuda-nvcc

python3 -c "import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"

#cd /mnt/c/apps/PycharmProjects/
mkdir -p $CONDA_PREFIX/etc/conda/activate.d
echo 'export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$CONDA_PREFIX/lib/' > $CONDA_PREFIX/etc/conda/activate.d/env_vars.sh
#重要,不然pycharm 运行tensorflow会找不到gpu依赖库

报错:
Node: ‘StatefulPartitionedCall_4’
2 root error(s) found.
(0) INTERNAL: libdevice not found at ./libdevice.10.bc
[[{{node StatefulPartitionedCall_4}}]]
[[crf/cond/StatefulPartitionedCall/crf/cond/else/_236/crf/cond/concat/_328]]
(1) INTERNAL: libdevice not found at ./libdevice.10.bc
[[{{node StatefulPartitionedCall_4}}]]
0 successful operations.
0 derived errors ignored. [Op:__inference_train_function_73705]
解决办法:

cp miniconda3/envs/tf2.12/nvvm /mnt/d/pyproject/myrasa2/ -r
#复制虚拟环境下的nvvm目录到项目根目录

相关文章:

  • 基于PyCATIA的CATIA实体双侧分割技术实现与优化
  • Onlyoffice 编译打包运行过程优化
  • 基于CNN的FashionMNIST数据集识别4——GoogleNet模型
  • VMware主机换到高配电脑,高版本系统的问题
  • 视图窗口的客户区
  • Android Coil3 Fetcher preload批量Bitmap拼接扁平宽图,Kotlin
  • 【VolView】纯前端实现CT三维重建-CBCT
  • Python中的优化函数5:优化的一些处理技巧
  • 如何选择合适的 AI 模型?(开源 vs 商业 API,应用场景分析)
  • 基于SpringBoot的“校园招聘网站”的设计与实现(源码+数据库+文档+PPT)
  • 从零开始驯服Linux(一):ZYNQ-Linux启动文件构建全解析
  • 【Linux】快速上手Makeflie CMake
  • 神聖的綫性代數速成例題10. N維矢量綫性運算、矢量由矢量組綫性表示、N個N維矢量相關性質
  • RC5解密工具
  • socks 协议介绍
  • 【Editor】动态添加/移除宏定义
  • css重点知识汇总(一)
  • 人工智能在2025年:各行业现状与变革
  • CMS漏洞-WordPress篇
  • 掌握 Shopee 商品数据:用爬虫解锁无限商机
  • 马斯克:大幅削减政治支出,仍将执掌特斯拉至少5年,除非去世
  • 太平人寿党委书记赵峰调任海南省政府党组成员
  • 黄仁勋:美国芯片管制完全错误,阻碍别人发展只会激励他们更努力
  • 痴情与真爱
  • 4名中学生同服处方药后身体不适,一心堂一药店未凭处方售药被罚1万元
  • 世卫大会再次拒绝涉台提案,国台办:民进党当局再遭挫败理所当然