【深度学习】Tensorflow/Pytorch GPU可运行环境记录

目的

每次重装tensorflow 或者 pytorch的时候总是要找不同环境下匹配的版本,在此特意记录一下可以运行的版本搭配。

记录

System NVIDIA驱动版本 CUDA版本 cudatoolkit cudnn tensorflow-gpu pytorch pytorch-cuda
Ubuntu 22.04 520.56.06 11.8 11.3.1 8.2.1 2.6.0 1.13.0 11.7
Ubuntu 22.04 525.60.11 12.0 11.3.1 8.2.1 2.6.0 1.13.0 11.7

欢迎大家在评论区提供可运行的环境搭配方案!我会定时添加进列表供大家参考!

你可能感兴趣的:(tensorflow,深度学习,pytorch,tensorflow)