安装Pytorch如何选择CUDA的版本
安装Pytorch时CUDA的选择
- Nvidia CUDA
- 查看CUDA版本方法
- 查看CUDA的驱动API版本
- 查看CUDA的运行API版本
- 查看官方CUDA的运行API版本
- 方法1
- 方法2
- 查看Anaconda里cudatoolkit包版本
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)
显示结果:
查看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。
方法2
输入下面的命令,也是显示官方提供的CUDA Toolkit版本。显示的版本和上面一样,也是7.5。
cat /usr/local/cuda/version.txt
查看Anaconda里cudatoolkit包版本
输入以下命令,在显示列表中找到cudatoolkit包就可以看到它的版本了。
conda list
显示的版本为10.2.89。这是我们运行pytorch的cuda版本,根据它来下载对应的pytorch即可。
以上为本人参考了博客和动手实践得出的结论,如有错误可在评论区指出,方便一起学习。
参考博客:
显卡,显卡驱动,nvcc, cuda driver,cudatoolkit,cudnn到底是什么?
CUDA、NVIDIA driver、多版本cuda
安装Pytorch如何选择CUDA的版本相关推荐
- win10 安装Pytorch GPU版+CUDA+cuDNN(篇幅较长保姆教程) (含CUDA11.1与cuDNN8.0.4安装包)
win10安装CUDA11.1+cuDNN8.0.4+Pytorch(含CUDA11.1与cuDNN8.0.4安装包) 更新NVIDIA(查看版本号) 查找NVIDIA,CUDA,cuDNN对应版本号 ...
- 安装pytorch 后torch.cuda.is_available()返回False的解决方法
今年安装pytorch 后一直只能安装上CPU版本,不能使用GPU. 参考了这篇博客的方法:完美解决torch.cuda.is_available()一直返回False的玄学方法(完美解决torch. ...
- 安装Pytorch后torch.cuda.is_available()返回False问题解决
首先说几点安装Pytorch时需要注意的点: 1.如果有能力,最好在科学上网后,利用类似如下命令直接安装: conda install pytorch torchvision torchaudio c ...
- window下安装pytorch(不用下载cuda和cudnn)(用清华镜像)
window下安装pytorch 前言: 最近换了一台笔记本电脑(3060显卡),单单为了配这个pytorch环境就花了1天时间(即使我以前配过).所以,现在想要记录一些细节防止忘记. 1.anaco ...
- ubuntu18.04安装pytorch、cuda、cudnn和miniconda
目录 0. 前言 1. nvidia驱动安装及卸载 1.1 安装 1.2 卸载 2. cuda-10.2安装验证及卸载 2.1 安装 2.2 验证 2.3 卸载 3. cudnn安装及验证 3.1 安 ...
- 安装pytorch gpu版本
在windows下用anaconda虚拟环境安装pytorch gpu版 成功! 文章目录 0. 用pip安装 1. 安装准备 2. 安装指令 3. 查看.提升cuda版本 3.1 判断显卡是否支持c ...
- Linux和Windows系统下:安装Anaconda、Paddle、tensorflow、pytorch,GPU[cuda、cudnn]、CPU安装教学,以及查看CPU、GPU内存使用情况
Linux和Windows系统下安装深度学习框架所需支持:Anaconda.Paddlepaddle.Paddlenlp.pytorch,含GPU.CPU版本详细安装过程 1.下载 Anaconda ...
- 稀里糊涂的解决了 cuda 和cudnn的安装以及conda安装pytorch出现的torch.cuda.is_available()为false的问题
参考的链接: https://blog.csdn.net/ashome123/article/details/105822040#commentBox Ubuntu 20.04 CUDA&c ...
- (最新最全)windows使用anaconda安装pytorch进行深度学习并使用GPU加速
本篇文章记录下自己安装pytorch的过程,由于我装过3~4次了,所以还算是比较有经验了. 文章目录 1.检查电脑配置 2.Anaconda的准备工作-添加国内镜像源 3.下载 3.测试 1.检查电脑 ...
最新文章
- 某单位招聘考试需要考核数学英语计算机,2010年10月自学考试电子商务网站设计原理试题...
- Cissp-【第6章 安全评估与测试】-2021-3-15(661页-706页)
- JLINK通过SW模式下载程序的方法
- Leetcode 167. 两数之和 II - 输入有序数组 解题思路及C++实现
- 【Android工具】Yandex!可以安装PCchrome插件的手机浏览器!更新网页剪辑插件测试情况...
- 蓝桥杯java第七届决赛第一题--愤怒小鸟
- 算法录 之 复杂度分析。
- 【时间序列】最完整的时间序列分析和预测(含实例及代码)
- python爬虫教程:爬虫的基本流程
- VueRouter源码详细解读
- golang 文件(文件打开,文件写入,文件读取,文件删除)的基本操作
- 工作389-移动端控制
- c++ python混合编程 restful_简单上手nodejs调用c++(c++和js的混合编程)
- 计算机操作员高级工试题一,计算机操作员高级工操作题复习资料
- django3,vue前后端分离数据请求
- JAVA面试--电商业内大厂
- day023 常用模块02
- Android-解决ViewFlipper与ScrollView滑动响应事件拦截的问题【转】
- 【BZOJ4837】LRU算法 [模拟]
- 1.编写一个程序,它从标准输入读取C源代码,并验证所有的花括号都正确的成对出现。
热门文章
- antlr4例子_ANTLR和网络:一个简单的例子
- flex布局水平垂直 垂直_垂直和水平装饰
- java ee maven_真正释放Maven和Java EE的强大功能
- nifty ui_Nifty JUnit:在方法和类级别上使用规则
- Spring-Boot 2.1.x和主要的bean定义
- ldap2.8.2_Spring LDAP 2.0.0发布
- 不知所措:您是否真的需要为您的API提供客户端库?
- eclipse pmd使用_使用您自己的规则在Eclipse中自定义PMD
- NetBeans Java EE技巧7:忽略的Java类和XHTML编辑器快捷方式
- jaxb 映射 空字段_JAXB和未映射的属性