WSL安装TensorFlow（with CUDA12.4）

下载并打开WSL

参考Windows下使用WSL设置启动WSL并安装Ubuntu版本分发
打开WSL

wsl -d Ubuntu

更新软件包

sudo apt-get update

sudo apt-get upgrade -y

安装配置CUDA 12.4

• 下载安装CUDA 12.4
  根据CUDA Toolkit 12.4NVIDIA官方安装参考
  按顺序输入以下命令即可

  wget https://developer.download.nvidia.com/compute/cuda/repos/wsl-ubuntu/x86_64/cuda-wsl-ubuntu.pin
  

  sudo mv cuda-wsl-ubuntu.pin /etc/apt/preferences.d/cuda-repository-pin-600
  

  wget https://developer.download.nvidia.com/compute/cuda/12.4.1/local_installers/cuda-repo-wsl-ubuntu-12-4-local_12.4.1-1_amd64.deb
  

  sudo dpkg -i cuda-repo-wsl-ubuntu-12-4-local_12.4.1-1_amd64.deb
  

  sudo cp /var/cuda-repo-wsl-ubuntu-12-4-local/cuda-*-keyring.gpg /usr/share/keyrings/
  

  sudo apt-get update
  

  sudo apt-get -y install cuda-toolkit-12-4
  

  rm cuda-repo-wsl-ubuntu-12-4-local_12.4.1-1_amd64.deb
  

• 配置CUDA环境变量
  • 打开用户配置文件

    vim ~/.bashrc
    

  • 点击键盘I键进入编辑模式
  • 将光标移动到最下面添加如下内容

    export CUDA_HOME