Windows10 linux子系统上安装CUDA|GPU加速|TENSORFLOW

GPU - CUDA - Windows Subsystem for Linux (WSL)

GPU in Windows Subsystem for Linux (WSL)
CUDA on Windows Subsystem for Linux (WSL)
https://developer.nvidia.com/cuda/wsl

CUDA on WSL User Guide

The guide for using NVIDIA CUDA on Windows Subsystem for Linux.

Changelog

• 6/19/2020: Updated driver release to address cache coherency issues on some CPU systems, including AMD Ryzen.

  The following software versions are supported with this preview release for WSL 2:
  • NVIDIA Driver for Windows 10: 455.41

• 6/17/2020: Initial Version.

  The following software versions are supported with this preview release for WSL 2:
  • NVIDIA Driver for Windows 10: 455.38
  • NVIDIA Container Toolkit: nvidia-docker2 (2.3) and libnvidia-container (>= 1.2.0-rc.1)

Introduction

Windows Subsystem for Linux (WSL) is a Windows 10 feature that enables users to run native Linux command-line tools directly on Windows. WSL is a containerized environment within which users can run Linux native applications from the command line of the Windows 10 shell without requiring the complexity of a dual boot environment. Internally, WSL is tightly integrated with the Microsoft Windows operating system, which allows it to run Linux applications alongside traditional Windows desktop and modern store apps.

Figure 1. CUDA on WSL Overview
 

With WSL 2 and GPU paravirtualization technology, Microsoft enables developers to run GPU accelerated applications on Windows.

The following document describes a workflow for getting started with running CUDA applications or containers in a WSL 2 environment.

Getting Started

Getting started with running CUDA on WSL requires you to complete these steps in order:

1. Installing the latest builds from the Microsoft Windows Insider Program
2. Installing the NVIDIA preview driver for WSL 2
3. Installing WSL 2

Installing Microsoft Windows Insider Program Builds

Install the latest builds from the Microsoft Windows Insider Program

• Register for the Microsoft Windows Insider Program.

• Install the latest build from the Fast ring.

  Note:

  Ensure that you install Build version 20145 or higher.

  You can check your build version number by running winver via the Windows Run command.

Installing NVIDIA Drivers

• Download the NVIDIA Driver from the download section on the CUDA on WSL page. Choose the appropriate driver depending on the type of NVIDIA GPU in your system - GeForce and Quadro.

• Install the driver using the executable. This is the only driver you need to install.

  Note:

  Do not install any Linux display driver in WSL. The Windows Display Driver will install both the regular driver components for native Windows and for WSL support.

Installing WSL 2

This section includes details about installing WSL 2, including setting up a Linux distribution of your choice from the Microsoft Store.

1. Install WSL 2 by following the instructions in the Microsoft documentation available here.
2. Ensure you have the latest kernel by hitting “Check for updates” in the “Windows Update” section of the Settings app. If the right update with the kernel 4.19.121+ is installed, you should be able to see it in the Windows Update history. Alternatively, you can check the version number by running the following command in PowerShell:

   wsl cat /proc/version
                       

    

   

   
    
3. If you don’t see this update, then in the Windows Update Advanced options, make sure to enable recommended Microsoft updates and run the check again: