1200字范文,内容丰富有趣,写作的好帮手!
1200字范文 > TensorFlow/PyTorch和cuda等版本对应关系

TensorFlow/PyTorch和cuda等版本对应关系

时间:2022-08-31 09:43:54

相关推荐

TensorFlow/PyTorch和cuda等版本对应关系

参考:

pytorch、显卡、显卡驱动、cuda版本是如何对应的:/p/ac70300b598b

NVIDIA系列显卡做深度学习,需要

在主机安装显卡驱动(driver)在docker中安装英伟达的科学计算库(cuda)在python中安装深度学习库(pytorch和TensorFlow等)

driver依赖显卡,cuda依赖driver,pytorch依赖cuda,于是就会有一些版本依赖问题

01 查看环境的方法(ubuntu)

参考:

Ubuntu下Pytorch, Tensorflow对应的Python、英伟达显卡驱动、CUDA、CUDNN版本与环境信息查看方法+安装老版本pytorch链接

查看python版本

python --version

查看CUDA版本

cat /usr/local/cuda/version.txt

查看GPU 型号

lspci | grep -i nvidia

查看NVIDIA驱动版本

sudo dpkg --list | grep nvidia-*

查看cudnn版本

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

查看Tensorflow版本

① 如果安装了pip或者conda可以运行

pip list

或者

conda list

找到其中Tensorflow查看版本

② 也可以进入python环境:

import tensorflow as tftf.__version__

查看Pytorch版本

① 安装了pip或者conda可以向上面一样运行

pip list

或者

conda list

找到其中叫做 torch 的包查看版本(注意不叫pytorch)

② 也可以进入python环境,输入:

import torchtorch.__version__

查看已经安装的Pytorch对应的CUDA版本

进入python环境,输入:

import torchtorch.version.cuda

查看已经安装的Pytorch对应的CUDNN版本

进入python环境,输入:

import torchtorch.backends.cudnn.version()

02 python与pytorch、cuda版本对应关系

/get-started/previous-versions/

03 python与TensorFlow、cuda版本对应关系

CPU版本的

/install/source_windows#cpu

GPU版本的

/install/source_windows#gpu

虽然官方是给的cuda10才可以TensorFlow1,但是下面这篇文章也使用了cuda10.1、python3.7和vs安装了tensorflow-gpu 1.14,可以参考一下~

/weixin_45023983/article/details/99178625

04 cuda和cudnn的对应版本

/rdp/cudnn-archive#a-collapse742-10

本内容不代表本网观点和政治立场,如有侵犯你的权益请联系我们处理。
网友评论
网友评论仅供其表达个人看法,并不表明网站立场。