首页 > 编程语言 >Python PyTorch GPU版/CUDA 环境详细安装教程(完美避坑)

Python PyTorch GPU版/CUDA 环境详细安装教程(完美避坑)

时间:2022-12-14 21:56:17浏览次数:68  
标签:Python Torch 避坑 PyTorch CUDA 版本 安装

一、安装 CUDA 

1、安装CUDA前请先确认Torch 所支持的版本,以免安装的Torch 不支持最新CUDA的版本,下图可见Torch所支持的CUDA最新版本位 11.7,如果我们安装CUDA 12.0 版本 执行 `torch.cuda.is_available()` 时则为False。

PyTorch版本查询链接: Start Locally | PyTorch

 

 

 

2、下载安装

根据上图所示,选择合适的CUDA版本,安装CUDA  C盘需要5 GB左右的空间,如果C盘空间少可使用分区助手来扩容,详细的安装过程不在概述,傻瓜试一键安装。

CUDA下载链接:CUDA Toolkit Archive | NVIDIA Developer

 二、安装PyTorch

 1、复制 `Run this Command` 这里的命令,在CMD中执行即可

 

标签:Python,Torch,避坑,PyTorch,CUDA,版本,安装
From: https://www.cnblogs.com/hlikex/p/16983707.html

相关文章

  • python之路49 模板层标签 自定义过滤器 模板继承、模型层准备、ORM部分操作
    模板层之标签{%if条件1(可以自己写也可以是用传递过来的数据)%}<p>周三了周三了</p>{%elif条件2(可以自己写也可以用传递过来的数据)%}<p>百日......
  • python学习笔记整理03(函数)
    1语法:1.1基本语法:#1.定义函数#使用def(define)关键字定义函数,且函数命名方法要遵循标识符规则deffunc1():#定义函数内的代码称为函数体print('函......
  • 用Python调用OpenAI API做有趣的事
    OverridetheentrypointofanimageIntroducedinGitLabandGitLabRunner9.4.Readmoreaboutthe extendedconfigurationoptions.Beforeexplainingtheav......
  • 仿真环境设置(一)- windows10, python 3.8
    1安装miniconda参考网上材料,打开Miniconda官网,点击对应系统版本的安装器,然后安装即可。安装过程中推荐:为所有人/只为我:只为我;将conda添加到PATH中,是/否:是。(虽然不......
  • Python笔记--字符串
    字符串的三种定义单引号定义法:双引号定义法:三引号定义法:结果:其中,三引号定义的话,不用变量接收的话,就相当于多行注释;用变量接收的话,就是对于字符串的定义了。字符串......
  • 【Python】 try except 还是 if else(EAFP 还是 LBYL 风格)
     结论● 若超过95%的可能不会进入except异常处理中,则使用代码一的样式(EAFP风格)● 否则可以换成用代码二的样式(LBYL风格) 代码一(EAFP风格):try:判断代码excep......
  • Python小球移动游戏
    #-*-coding:utf-8-*-importsys#导入sys模块importpygame#导入pygame模块pygame.init()#初始化pygamesize=width,height=640,480#设置窗口screen......
  • Python 12章 GUI
    GUI界面此处使用wxpython需要先pip库实例1importwxclassMyFrame(wx.Frame):def__init__(self,parent,id):wx.Frame.__init__(self,parent,id,tit......
  • Python实现从一个列表数据里随机抽取数据,并且按原有顺序排序
    背景:工作中需要实现从多个条件中随机抽取几个条件,进行组合查询的功能。而查询的结果需要按原顺序进行判断是否符合查询条件。分析:这些条件可以放在列表里,这就需要实现一个......
  • Python13 模块
    模块创建模块直接新建一个.py文件即可,这个py文件是可执行的哦~也是可以被导入的,这个py文件就是模块。导入模块import模块名[as别名]from模块名import函数/变......