首页 > 其他分享 >WSL2 配置 tensorflow 环境

WSL2 配置 tensorflow 环境

时间:2024-03-18 16:12:31浏览次数:23  
标签:libcuda lib 配置 wsl so.1 usr tensorflow WSL2 CUDA

Windows 系统中更新NVIDA驱动

这里可以直接通过 Geforce Experience 直接更新
更新完成后可以在命令行/wsl中输入 nvidia-smi

可以看到输出

这里的CUDA Version指的是该驱动版本最高可支持的CUDA版本

安装CUDA

NVIDIA官网下载符合条件的CUDA

这里我一开始直接选择安装了最新版的CUDA 12.4,随后发现Tensorflow 目前(2024.3.17) 并不支持 12.4,于是重新安装,选择了CUDA 11.2



直接下载EXE版本安装即可

在安装时,会发现安装报错 cuda you already have a newer version of the nvidia frameview sdk installed

这里需要到系统设置里卸载NVIDIA frameview sdk 组件,我使用的是Geek进行卸载,很干净

注意:不要直接到NVIDIA 文件夹删除对应文件

下载ANACONDA

最新下载地址找到对应的ANACONDA版本,复制链接

https://www.anaconda.com/products/distribution#linux
wget https://repo.anaconda.com/archive/Anaconda3-2021.11-Linux-x86_64.sh
bash ./Anaconda3-2021.11-Linux-x86_64.sh

根据脚本说明安装并编辑环境变量

export PATH="/path/to/conda/bin:$PATH"

使用conda init进行初始化

WSL中下载CUDA TOOLKIT

前往官网找到对应版本的TOOLKIT:https://developer.nvidia.com/cuda-toolkit-archive

复制官网指令进行安装

注意:如果不是最新版本,需要将最后一行修改为对应的版本号

错误处理

这里如果之前安装过不同版本的TOOLKIT,可能会报错

Errors were encountered while processing:
 /tmp/apt-dpkg-install-vFgFO4/00-cuda-cudart-11-2_11.2.72-1_amd64.deb
E: Sub-process /usr/bin/dpkg returned an error code (1)

这里需要卸载已有的CUDA TOOLKIT

   sudo dpkg --force-all -P $(dpkg -l | grep cuda | awk '{print $2}')

发现报错:

Removing cuda-toolkit-12-4-config-common (12.4.99-1) ...
Removing alternatives
Purging configuration files for cuda-toolkit-12-4-config-common (12.4.99-1) ...
Removing alternatives
Processing triggers for libc-bin (2.31-0ubuntu9.14) ...
/sbin/ldconfig.real: /usr/lib/wsl/lib/libcuda.so.1 is not a symbolic link

运行 ls -l /usr/lib/wsl/lib/libcuda.so*

发现对应文件不是一个符号链接

-r-xr-xr-x 1 root root 162552 Mar  2 07:04 /usr/lib/wsl/lib/libcuda.so
-r-xr-xr-x 1 root root 162552 Mar  2 07:04 /usr/lib/wsl/lib/libcuda.so.1
-r-xr-xr-x 1 root root 162552 Mar  2 07:04 /usr/lib/wsl/lib/libcuda.so.1.1

删除错误的 libcuda.so.1 文件:

sudo rm /usr/lib/wsl/lib/libcuda.so.1

创建一个新的符号链接,使 libcuda.so.1 指向具体版本的文件,如 libcuda.so.1.1

sudo ln -s /usr/lib/wsl/lib/libcuda.so.1.1 /usr/lib/wsl/lib/libcuda.so.1

运行 ldconfig 来更新库的缓存信息:

sudo ldconfig

使用

sudo apt-get autoremove
sudo apt-get autoclean

清理本地存储库里的软件包信息

重新安装即可

安装cuDNN

还是一样,找到对应的安装文件:https://developer.nvidia.com/rdp/cudnn-archive,这里需要注册登录后才能下载

使用windows下载完成后移动到Ubuntu系统中。WLS2中:

  • Linux 文件系统被映射到\wsl$\Ubuntu-20.04\
  • Windows的磁盘被挂载到了/mnt下,可以直接访问

完成后通过sudo dpkg -i NAME即可安装

使用conda建立对应版本的python虚拟环境

 conda create --name myenv python=3.8
 conda activate myenv

之后即可使用对应指令安装对应版本的TensorFlow

pip install tensorflow-gpu==2.11.0

Jupyter

pip install jupyterlab

使用如下命令启动jupyter lab:jupyter lab --no-browser

测试tensorflow

import tensorflow as tf
tf.config.list_physical_devices()

发现报错

在windows桌面右键打开NVIDIA控制面板

将PhysX设置修改为指定显卡

成功解决

标签:libcuda,lib,配置,wsl,so.1,usr,tensorflow,WSL2,CUDA
From: https://www.cnblogs.com/stypro/p/18080643

相关文章

  • 【黑马MySQL】MySQL的下载&安装&启停&配置环境变量【一条龙教程】
    前言大家好吖,欢迎来到YY滴MySQL系列,热烈欢迎!本章主要内容面向接触过C++Linux的老铁主要内容含:欢迎订阅YY滴C++专栏!更多干货持续更新!以下是传送门!YY的《C++》专栏YY的《C++11》专栏YY的《Linux》专栏YY的《数据结构》专栏YY的《C语言基础》专栏YY的《初学者易......
  • Tomcat安装与配置详细教程:从入门到精通
    Tomcat安装与配置详细教程:从入门到精通简介:本教程旨在为广大开发者提供一份Tomcat服务器的安装与配置指南。通过本教程的学习,您将能够掌握Tomcat服务器的安装步骤、环境变量的配置方法,以及验证Tomcat配置是否成功的技巧。同时,我们还将简要介绍JavaJDK的安装与配置,为Tomca......
  • 配置全局变量直接调用函数
    配置全局变量直接调用函数目的想要在项目中使用defs.utills.getFunction()的方式,直接调用我们的函数,不再使用import的方式进行导入,直接在工程化上做手脚进行自动导入解决。过程首先我们在我们的项目框架src/目录下定义一个全局的utill文件,这个我们可以分类型创建文件夹、......
  • 【Java】i18n国际化解决方案:通过AOP切面实现多语言的配置
    需求背景国际化多语言配置。相较于常规的方法,这次采取了切面的方式,来完成所有字段->不同语言的映射。大致逻辑:按常规的国际化,写一个获取语言的方法:getMsg(Stringcode);写一个深层遍历对象的方法traverseObject(Objectobj),通过反射,获取所有类型为字符串的字段,并实现对字......
  • ubuntu系统Jenkins + Gitlab 集成 实现CI\CD (配置前后端)
    官网地址:Linux首先配置jdk17的环境,jenkins依附于java安装gitaptinstallgit更新aptupdate安装jdk17aptinstallopenjdk-17-jdkvim/etc/profileexportJAVA_HOME=/usr/local/java/exportJRE_HOME=$JAVA_HOME/jreexportCLASSPATH=$JAVA_HOME/lib:$JRE_HOME/lib......
  • linux 服务器互信配置
    [root@bre01opt]#catenv.csvippasswdhostname192.168.248.129123123bre01192.168.248.130123123bre02[root@bre01opt]#catenv.sh#!/bin/bashfunctionRSA(){echo"设置主机公钥">>./install_init.logrpm-qa|grepexpectif[$?-eq0]......
  • 【附下载】3Ds Max从安装、配置到入门提高和高级用法
    #3DsMax一、安装1.1安装说明地址:链接:https://pan.baidu.com/s/1lwKMbgbE32wCL6PpMv706A?pwd=dll8提取码:dll8–来自百度网盘超级会员V2的分享安装说明:文件夹里有安装说明安装解压即可关键就是将crack文件放到自己的安装目录即可1.2安装小技巧尽量新建一个一个......
  • linux配置文件
    [.bashrc]bashresourceconfiguration配置文件[给命令设置别名]设置临时别名,只在当前终端有效//设置别名user$aliaslss='ls'//查看已经设置的别名user$alias//删除别名user$unaliaslss设置永久别名,在~/.bashrc中间中写入alias别......
  • docker配置文件解析
    {"authorization-plugins":[],//用于指定授权插件的列表,一般用于自定义认证"data-root":"",//Docker运行时使用的根路径,默认/var/lib/docker"dns":[],//设定容器DNS的地址,在容器的/etc/resolv.conf文件中可查看"dns-opts":[],//容器/etc/resolv.conf文件,其他设......
  • VSCODE安装GO语言并配置go module启动
    前言go语言近期太火了,尤其是公司需要快速搭建服务上线,非常推荐大家学习,而且这个语言写法跟前端的ts非常相似,因此继php之后,nodejs+ts转go也是上手非常快。安装成功后查看版本goversion配置gomodule1goenv-wGO111MODULE=on2goenv-wGOPROXY=https://gopro......