首页 > 其他分享 >安装Pytorch如何选择CUDA的版本,看这一篇就够了

安装Pytorch如何选择CUDA的版本,看这一篇就够了

时间:2023-02-21 09:59:02浏览次数:48  
标签:Toolkit 就够 Pytorch API CUDA 版本 运行 cuda

CUDA是一个并行计算平台和编程模型,能够使得使用GPU进行通用计算变得简单和优雅。Nvidia官方提供的CUDA 库是一个完整的工具安装包,其中提供了 Nvidia驱动程序、开发 CUDA 程序相关的开发工具包等可供安装的选项。

Nvidia CUDA

Nvidia安装的cuda是一个完整的cuda库,包含cuda toolkit和相应的驱动。主要包含三部分:CUDA toolkit、CUDA driver和NVIDIA GPU driver。理论上在安装完驱动以后只需要安装CUDA toolkit就可以保证CUDA相关的程序运行。

查看CUDA版本方法

CUDA分为两种,驱动API和运行API,驱动API指的是指的显卡驱动支持的最高cuda版本,我们运行程序时用的是运行API。

查看CUDA的驱动API版本

表示的是驱动所能支持的最大运行API版本。如果显示11.3,用我们常说的话就是最大支持11.3的cuda。下面为查看CUDA驱动API版本的命令:

nvidia-smi  #右上角显示CUDA版本(驱动API)

显示结果:
image

查看CUDA的运行API版本

这是平时我们所说的CUDA版本,由于运行API在CUDA里的CUDA Toolkit工具包中,所以运行API版本也是CUDA Toolkit工具包的版本。其实装了Anaconda之后Anaconda也会提供一个cudatoolkit工具包,同样包含了CUDA的运行API,可以用来替代官方CUDA的CUDA Toolkit。这也就是为什么有时候我们通过nvcc-V查看的cuda版本很低(比如7.5),但是能成功运行cuda9.0的pytorch的原因。因为在安装完anaconda后,运行pytorch代码就会使用anaconda的cudatoolkit,而忽视官方的CUDA Toolkit,所以我们只需要根据anaconda的cudaoolkit包的版本来安装相应的pytorch即可。下面为查看运行API版本的命令:

查看官方CUDA的运行API版本

方法1

nvcc -V

结果显示官方CUDA提供的CUDA Toolkit工具包的版本为7.5。
image
方法2
输入下面的命令,也是显示官方提供的CUDA Toolkit版本。显示的版本和上面一样,也是7.5。

cat /usr/local/cuda/version.txt

查看Anaconda里cudatoolkit包版本
输入以下命令,在显示列表中找到cudatoolkit包就可以看到它的版本了。

conda list

image
显示的版本为10.2.89。这是我们运行pytorch的cuda版本,根据它来下载对应的pytorch即可。

标签:Toolkit,就够,Pytorch,API,CUDA,版本,运行,cuda
From: https://www.cnblogs.com/chaofengya/p/17139887.html

相关文章

  • pytorch中学习率的调整方法
    一、手动法二、利用lr_scheduler()提供的集中衰减函数2.1利用lr_lambda函数具体使用:fromtorch.optimimportSGD,lr_schedulerimportmatplotlib.pyplotaspltfro......
  • 图卷积神经网络分类的pytorch实现
    图神经网络(GNN)目前的主流实现方式就是节点之间的信息汇聚,也就是类似于卷积网络的邻域加权和,比如图卷积网络(GCN)、图注意力网络(GAT)等。下面根据GCN的实现原理使用Pytorch......
  • 使用PyTorch-LSTM进行单变量时间序列预测的示例教程
    时间序列是指在一段时间内发生的任何可量化的度量或事件。尽管这听起来微不足道,但几乎任何东西都可以被认为是时间序列。一个月里你每小时的平均心率,一年里一只股票的日收......
  • 深度学习框架百度飞桨、Facebook pytorch和谷歌 TensorFlow的区别
    百度飞桨、Facebookpytorch和谷歌TensorFlow的区别https://www.zhihu.com/question/449784762/answer/2620586949【window环境】Python安装与深度学习环境(tensorflow)配......
  • PyTorch 并行训练 DistributedDataParallel完整代码示例
    使用大型数据集训练大型深度神经网络(DNN)的问题是深度学习领域的主要挑战。随着DNN和数据集规模的增加,训练这些模型的计算和内存需求也会增加。这使得在计算资源有......
  • CUDA 教程(一) GPU 编程概述和 CUDA 环境搭建
    前言 本文将带大家了解CUDA编程,学习使用C和Python调用CUDA编程接口,对模型进行推理加速,性能优化。我们将从底层原理开始,带大家了解CUDA的工作机制,一步步掌握CUDA编程。相......
  • java处理异常这一篇就够了
    java异常packagecom.text01;importjava.util.Scanner;publicclassText{publicstaticvoidmain(String[]args){//实现一个功能://键盘录入俩个......
  • Pytorch 看起来好像没占gpu的样子的原因
    今天好哥们儿赞助的3080到手了,欣喜若狂的装上然后跑了跑MNIST,看着任务管理器CPU跑100%,GPU跑3%,查了半天解决不了,郁闷了好一会儿。。 后来在https://www.bilibili.com/r......
  • (Pytorch第三天)复制官网教程程序遇到的一系列报错
    先说一下,我之前没学过Python,但是鉴于我是一名c语言高手(雾),我决定现学现用。首先是https://pytorch.org/tutorials/beginner/basics/optimization_tutorial.html的  位......
  • Windows10下YOLOv8 TensorRT CUDA加速部署
    前言本文介绍了Windows10下YOLOv8TensorRTCUDA加速部署。 欢迎关注公众号CV技术指南,专注于计算机视觉的技术总结、最新技术跟踪、经典论文解读、CV招聘信息。 ......