pytorch，cuda，cudatoolkit，driver版本详解

pytorch，cuda，cudatoolkit，driver四者关系如下：

系统的Nvidia Driver决定着系统最高可以支持什么版本的cuda和cudatoolkit，Nvidia Driver是向下兼容的，详情如下（见Nvidia Driver和Cuda对应关系）
cuda和cudatoolkit不同，前者说的是系统安装的cuda，它是由Nvidia官方提供的（/usr/local/cuda就是系统安装的cuda的软链接），这与我们要安装的pytorch几乎没有什么关系。后者是anaconda官方提供的用来build pytorch的一个工具包，它是Nvidia所提供的cuda的一个子集。
pytorch和cudatoolkit版本并不是一一对应的关系，一个pytorch版本可以有多个cudatoolkit版本与之对应。例如1.5.1版本的pytorch，既可以使用9.2版本的cudatoolkit，也可以使用10.2版本的cudatoolkit。

pytorch安装的坑

只指定pytorch版本来安装不一定是能work的，例如执行conda install pytorch=X.X.X -c pytorch时，conda会自动为你选择合适版本的cudatoolkit。但conda只能保证你的pytorch和cudatoolkit版本一定是对应的，但并不能保证pytorch可以正常使用，因为系统的Nvidia Driver有可能不支持你所安装的cudatoolkit版本。

所以，除非你对你的Nvidia driver版本很有自信，否则，还是先查看系统Nvidia Driver的版本，并在上方图表中查询最高支持的cudatoolkit版本，然后指定cudatoolkit版本来安装pytorch吧。例如系统的Nvidia Driver版本为440.33，查询到最高支持cudatoolkit版本为10.2，则可以使用conda install pytorch cudatoolkit=10.2 -c pytorch命令安装pytorch。

当然，如果你对pytorch版本有特别的要求，你可以同时指定pytorch和cudatoolkit的版本。如果这两个版本不能兼容，系统会报错，例如：

conda install pytorch=1.5.1 cudatoolkit=9.0 -c pytorch

conda会告诉你：

Solving environment: failedUnsatisfiableError: The following specifications were found to be in conflict:- cudatoolkit=9.0 -> __cuda[version='>=9.0']- pytorch=1.5.1
Use "conda info <package>" to see the dependencies for each package.

如果你的Nvidia Driver最高支持的cudatoolkit版本为9.0，而你又一定要用1.5.1版本的pytorch，那么你必须升级系统的Nvidia Driver版本。

pytorch，cuda，cudatoolkit，driver版本详解相关推荐

Pytorch|YOWO原理及代码详解(二)
Pytorch|YOWO原理及代码详解(二) 本博客上接,Pytorch|YOWO原理及代码详解(一),阅前可看. 1.正式训练 if opt.evaluate:logging('evaluating ...
Pytorch | yolov3原理及代码详解（二）
阅前可看: Pytorch | yolov3原理及代码详解(一) https://blog.csdn.net/qq_24739717/article/details/92399359 分析代码: ht ...
时间序列预测任务PyTorch数据集类——TimeSeriesDataSet 类详解
时间序列预测任务PyTorch数据集类--TimeSeriesDataSet 类详解当进行时间序列预测或时间序列分析时,通常需要对数据进行预处理和转换以提高模型的效果和准确性.TimeSeriesD ...
Apache Hadoop版本详解
由doop版本混乱多变,因此,Hadoop的版本选择问题一直令很多初级用户苦恼.本文总结了ApacheHadoop和Cloudera Hadoop的版本衍化过程,并给出了选择Hadoop版本的一些建议 ...
eclipse版本详解以及下载安装步骤及启动eclispe报错整理
一.eclispe版本详解本段内容转载自:https://www.cnblogs.com/chenmingjun/p/8404628.html 1.eclipse版本代号一览表 Eclipse的设计 ...
rpm安装mysql5.6_利用rpm安装mysql 5.6版本详解
前言其实之前使用yum安装MySQL确实很方便,但是默认安装的myql5.0版本的,不支持utf8mb4(utf8mb4扩展到一个字符最多能有4节,所以能支持更多的字符集,比如支持emoji表情)编 ...
CUDA: Occupancy(占用率)详解
#CUDA: Occupancy(占用率)详解占用率是指每个多处理器(Streaming Multiprocessor,SM)的活动线程束(warps)数量与实际的活动warps数量的比率. 高的占 ...
【Pytorch】torch.argmax 函数详解
文章目录一.一个参数时的 torch.argmax 函数 1. 介绍 2. 实例二.多个参数时的 torch.argmax 函数 1. 介绍 2. 实例实例1:二维矩阵实例2:三维矩阵实例3 ...
PyTorch入门笔记-matmul函数详解
PyTorch入门笔记-matmul函数详解本文转载自:PyTorch入门笔记-matmul函数详解 - 腾讯云开发者社区-腾讯云 (tencent.com) 41409)]

pytorch，cuda，cudatoolkit，driver版本详解

pytorch安装的坑

pytorch，cuda，cudatoolkit，driver版本详解相关推荐

最新文章

热门文章