TensorFlow/PyTorch和cuda等版本对应关系
参考:
- pytorch、显卡、显卡驱动、cuda版本是如何对应的:https://www.jianshu.com/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 tf
tf.__version__
查看Pytorch版本
① 安装了pip或者conda可以向上面一样运行
pip list
或者
conda list
找到其中叫做 torch 的包查看版本(注意不叫pytorch)
② 也可以进入python环境,输入:
import torch
torch.__version__
查看已经安装的Pytorch对应的CUDA版本
进入python环境,输入:
import torchtorch.version.cuda
查看已经安装的Pytorch对应的CUDNN版本
进入python环境,输入:
import torch
torch.backends.cudnn.version()
02 python与pytorch、cuda版本对应关系
https://pytorch.org/get-started/previous-versions/
03 python与TensorFlow、cuda版本对应关系
CPU版本的
https://tensorflow.google.cn/install/source_windows#cpu
GPU版本的
https://tensorflow.google.cn/install/source_windows#gpu
虽然官方是给的cuda10才可以TensorFlow1,但是下面这篇文章也使用了cuda10.1、python3.7和vs2019安装了tensorflow-gpu 1.14,可以参考一下~
https://blog.csdn.net/weixin_45023983/article/details/99178625
04 cuda和cudnn的对应版本
https://developer.nvidia.com/rdp/cudnn-archive#a-collapse742-10
TensorFlow/PyTorch和cuda等版本对应关系相关推荐
- window下TensorFlow CUDA,CUDNN版本对应关系
- tensorflow对应的python版本_详解Tensorflow不同版本要求与CUDA及CUDNN版本对应关系
参考官网地址: Windows端:https://tensorflow.google.cn/install/source_windows CPU Version Python version Comp ...
- python和tensorflow版本对应_详解Tensorflow不同版本要求与CUDA及CUDNN版本对应关系
参考官网地址: CPU Version Python version Compiler Build tools tensorflow-1.11.0 3.5-3.6 MSVC 2015 update 3 ...
- tensorflow 安装GPU版本,CUDA与cuDNN版本对应关系,RTX3050Ti (notebook)
前言 安装Tensorflow-gpu 与 keras的时候,一定先要注意版本的对应,不然很容易出错,在看的时候,建议先看完整篇文章再上手. 一.环境+配置 本机环境 显卡:RTX3050Ti(not ...
- centos GPU tensorflow pytorch 深度学习 环境搭建
简要记录一下深度学习开发环境的搭建步骤,懒得编辑了,将就着吧. 总体流程参考文章:https://www.cnblogs.com/learn-the-hard-way/p/12318980.html ...
- Cuda与cudnn对应关系
Tensorflow 与Cuda.cudnn对应关系 Cuda和cudnn是支持NVIDIA支持GPU的两个库,分别用于高性能计算和深度神经网络计算的支持. Cuda(Compute Unified ...
- (win10)anaconda虚拟环境中安装特定版本cuda+tensorflow+pytorch+pycharm
1.安装anaconda 在官网安装最新版的anaconda,需要使用其他版本的python可创建虚拟环境.安装时选择all users以及添加环境变量. 2.安装cuda以及cudnn. 你可以选择 ...
- 记录-安装cuda与cudnn 及对应版本的tensorflow|pytorch
仅用以记录个人环境配置流程. 一.安装CUDA Driver 1.1 查看CUDA Driver版本 1.2 驱动更新下载地址:NVIDIA Studio 驱动程序 | 512.59 | Window ...
- tensorflow 各个版本的 CUDA 以及 Cudnn 版本对应关系
本文由 简悦 SimpRead 转码, 原文地址 blog.csdn.net tensorflow 各个版本的 CUDA 以及 Cudnn 版本对应关系 一.tensorflow 各个版本需要的 CU ...
最新文章
- 西安邮电大学计算机学院系主任,西安邮电大学计算机学院
- 最高10万奖励!不限选题!高通AI应用创新大赛应用赛道详解
- 探索感染了COVID-19的动物的数据
- LeetCode-316. 去除重复字母(hard且重要)
- firefly 环境配置所需工具
- 回文数 Exercise06_03
- udpSocket捕获ICMP不可达错误
- 2020-11-25
- 2019FME博客大赛——基于FME的武汉市航班数据获取及城市联系度分析
- React的CRA脚手架的创建
- 函数指针数组的一些用法
- css箭头图片方向转换
- 2021 ICCV论文分享 | 遮挡边界检测
- 华为苏箐被曝转投大众汽车:最狂智能车高管,因抨击特斯拉丢工作,任正非亲自签发罢免令...
- 八百元八核的服务器?二手服务器(工作站)搭建指南(下)
- 根轨迹的基本概念与绘制
- java ajax 输出_正本清源:所谓Ajax输出的三种形式
- 小巧灵便的桌面工具带有便签、闹钟、任务管理功能
- 社保卡(深圳)在线激活
- (转)日志管理软件logstash