Linux下PyTorch、CUDA Toolkit 及显卡驱动版本对应关系(附详细安装步骤)

Linux下PyTorch、CUDA Toolkit 及显卡驱动版本对应关系(附详细安装步骤)Linux下PyTorch、CUDAToolkit及显卡驱动版本对应关系(附详细安装步骤)

大家好,又见面了,我是你们的朋友全栈君。

1.CUDA驱动和CUDA Toolkit对应版本


表一:CUDA驱动及CUDA Toolkit最高对应版本

在这里插入图片描述

最新可查阅官方文档

注:驱动是向下兼容的,其决定了可安装的CUDA Toolkit的最高版本。

2.CUDA Toolkit版本及其可用PyTorch对应版本(参考官网,欢迎评论区补充)


表二:CUDA Toolkit版本及可用PyTorch对应关系

CUDAToolkit版本 可用PyTorch版本
7.5 0.4.1 ,0.3.0, 0.2.0,0.1.12-0.1.6
8.0 1.1.0,1.0.0 ,0.4.1
9.0 1.1.0,1.0.1, 1.0.0,0.4.1
9.2 1.7.1,1.7.0,1.6.0,1.5.1,1.5.0,1.4.0,1.2.0,0.4.1
10.0 1.2.0,1.1.0,1.0.1 ,1.0.0
10.1 1.7.1,1.7.0,1.6.0,1.5.1,1.5.0, 1.4.0,1.3.0
10.2 1.11.0,1.9.0,1.8.1,1.8.0,1.7.1,1.7.0,1.6.0,1.5.1,1.5.0
11.0 1.7.1,1.7.0
11.1 1.8.0
11.3 1.11.0,1.9.0,1.8.1

注:虽有的卡驱动更新至较新版本,且CUDA Toolkit及PyTorch也可对应更新至新版本。但有的对应安装包无法使用,有可能是由于卡太旧的原因。

3.安装指导

在安装时会同时安装CUDA Toolkit以及PyTorch,这是我们要知道的。

步骤一:

使用nvidia-smi查询驱动版本:
在这里插入图片描述
如图中Driver Version所示,该卡目前的驱动版本为384.81

步骤二:
此处提供三种方法可供选择。
(1)指定CUDA Toolkit版本(推荐
根据表一查询到可安装的CUDA Toolkit版本,384.81对应最高的CUDA Toolkit版本为9.0
运行conda install pytorch cudatoolkit=9.0 -c pytorch即可。
此方法指定CUDA Toolkit版本后,conda会自动匹配到合适版本的PyTorch。

(2)指定PyTorch版本
根据表一查询到可安装的CUDA Toolkit版本,再根据表二查询到合适版本的PyTorch。384.81对应最高的CUDA Toolkit版本为9.0,9.0可安装PyTorch1.1.0版本。
运行conda install pytorch=1.1.0 -c pytorch即可。
此方法指定PyTorch版本后,conda会自动匹配到合适版本的CUDA Toolkit。

(3)同时指定CUDA Toolkit版本和PyTorch
根据表一查询到可安装的CUDA Toolkit版本,根据表二查询到合适版本的PyTorch
运行conda install pytorch=1.1.0 cudatoolkit=9.0 -c pytorch即可。

注:
(1)需要安装torchaudio以及torchvision时将其放在pytorch之后即可,如:conda install pytorch torchaudio torchvision cudatoolkit=9.0 -c pytorch
(2)-c pytorch为采用官网源下载,如果去掉,则在清华镜像源配置后可采用清华镜像源下载,配置清华镜像源可参考:Linux下conda使用清华镜像源快速安装PyTorch [CSDN]
(3)PyTorch1.8.0,1.9.0及1.0.0以前版本使用conda安装时命令有些许不同,具体可查看官网

4.验证安装是否成功

#使用python运行
import torch
print(torch.__version__)
print(torch.cuda.is_available())

卸载当前版本PyTorch:

conda uninstall pytorch

推荐阅读
Linux下conda使用清华镜像源快速安装PyTorch [CSDN]
conda安装环境相关 [CSDN]
PyTorch旧版本和CUDA Tookit版本对应情况(可复制对应安装命令)
PyTorch官网(下载最新版)

参考:
PyTorch [官网]
Pytorch如何更新版本与卸载,使用pip,conda更新卸载Pytorch [CSDN]

版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 举报,一经查实,本站将立刻删除。

发布者:全栈程序员-用户IM,转载请注明出处:https://javaforall.cn/149169.html原文链接:https://javaforall.cn

【正版授权,激活自己账号】: Jetbrains全家桶Ide使用,1年售后保障,每天仅需1毛

【官方授权 正版激活】: 官方授权 正版激活 支持Jetbrains家族下所有IDE 使用个人JB账号...

(0)


相关推荐

  • php js页面刷新,JS刷新页面方法总结

    php js页面刷新,JS刷新页面方法总结这次给大家带来JS刷新页面方法总结,JS刷新页面的注意事项有哪些,下面就是实战案例,一起来看一下。JS刷新当前页面的几种方法总结reload方法,该方法强迫浏览器刷新当前页面replace方法,该方法通过指定URL替换当前缓存在历史里(客户端)的项目,因此当使用replace方法之后,你不能通过“前进”和“后退”来访问已经被替换的URL返回并刷新页面自动刷新页面的方法js刷新页面方法大全页面自…

  • 管理系统源码下载_资产管理系统源码

    管理系统源码下载_资产管理系统源码spring源码mybatis源码hibernate源码struts源码下载spring源码下载地址:http://www.maixiaogang.commybatis源码下载地址:https://github.com/mybatis/mybatis-3/releaseshibernate源码下载地址:http://hibernate.org/orm/downloads/strut…

  • 大数据教程,大数据学习线路图

    大数据教程,大数据学习线路图前言先引用一下马云大大的话:很多人还没搞清楚什么是PC互联网,移动互联网来了,我们还没搞清楚移动互联的时候,大数据时代又来了。马云深度解析大数据“大数据”是近年来IT行业的热词,并广泛的应用在各行各业。特别是近年来随着社交网络、物联网、云计算以及多种传感器的广泛应用,以数量庞大,种类众多,时效性强为特征的非结构化的数据不断涌现,数据的重要性愈发凸显,传统的数据存储、分析技术难以实时处…

  • 几何画板 很实用的一个软件

    几何画板 很实用的一个软件

  • 条件运算符用法「建议收藏」

    条件运算符用法「建议收藏」别人的书上看到一个简单题目,刚开始学C的小伙伴们应该很熟悉,还记得当年在软件实验室那会儿,入学考试全是这个,做到吐。打印图形以下图形:* *** ***** ******* ********* *********** ************* *********** ********* ******* ***** *** *  一般方法:两重循环,里面是两…

  • struts2原理及创建struts工程步骤

    struts2原理及创建struts工程步骤

发表回复

您的电子邮箱地址不会被公开。

关注全栈程序员社区公众号