在WSL2上安装Tensorflow-GPU

Question

是否有人成功地在WSL2上安装了Tensorflow-GPU？我在WSL2上有Ubuntu18.04，但我很难安装NVIDIA驱动程序。如果我迷路了，任何帮助都会很感激的。

Answer 1

我可以确认我能够在不需要Docker的情况下在WSL2上工作，这要感谢以下文章：

https://qiita.com/Navier/items/cf551908bae707db4258

一定要更新到驱动程序版本460.15，而不是455.41，如CUDA文档所列。

注意，这不适用于TCC模式下的卡(只有WDDM)。另外，一定要将文件放在Linux文件系统上(例如，不要放在挂载驱动器上，比如/mnt/c/)。在Linux文件系统上，性能要快得多(这与WSL 1与WSL 2的实现不同有关；参见1、2和3.)。

注：也见Keras/Tensorflow中的类生成器(继承序列)线程安全吗？

Answer 2

这些是我在Ubuntu20.04中必须遵循的步骤。我已经不在开发通道上了，beta通道在这个用例中工作得很好，而且更稳定。

安装WSL2

从Windows商店安装Ubuntu 20.04

从：https://developer.nvidia.com/cuda/wsl/download安装Windows驱动程序

使用：sudo apt install nvidia-cuda-toolkit在WSL内部安装nvcc

检查它是否在那里：nvcc --version

对于我的用例，我做数据科学，并且已经安装了anaconda。我创建了一个环境：

conda create --name tensorflow
conda install tensorflow-gpu

然后用这个激活环境的python程序来测试它：

import tensorflow as tf
tf.config.list_physical_devices('GPU')
sys_details = tf.sysconfig.get_build_info()
cuda = sys_details["cuda_version"]
cudnn = sys_details["cudnn_version"]
print(cuda, cudnn)

由于我不明白的原因，我的机器找不到GPU而没有安装nvcc，实际上给出了一个错误消息，说它找不到nvcc。

在线教程，我已经发现，其中有您下载的CUDA和CUDNN分别，但我thinkNVCC包括CUDNN，因为它是。。。不知怎么的。

Answer 3

我只想指出，使用anaconda安装cudatoolkit和cudnn在wsl中似乎不起作用。

也许路径存在一些问题，使得TF只能在系统路径中而不是在conda环境中查找所需的文件。