首页 > 其他分享 >CUDA与Pytorch安装

CUDA与Pytorch安装

时间:2024-03-31 16:13:16浏览次数:22  
标签:Pytorch pytorch CUDA 版本 显卡 安装 下载 cuda

cuda和pytorch是使用python进行深度学习常会需要的工具,其中pytorch是深度学习的框架之一,cuda是利用GPU进行运算的工具。

  • cuda的安装
    cuda是英伟达公司开发的利用显卡进行深度学习的工具。显卡的GPU比CPU的运算能力要强,在深度学习时算力十分重要,直接决定了我们训练模型的速度,所以一般都会用到。但是cuda的下载经常出现一些问题,因此记录下来方便后续查看。
  1. 查看电脑支持的cuda版本
    由于不同的显卡能支持的cuda版本是不一样的,需要先确认版本。首先对显卡驱动进行更新,然后打开命令行,在其中输入Nvidia-smi查看显卡支持的cuda版本。

    可以看到我的电脑支持12.1版本的cuda。也可以通过英伟达控制面板,在帮助-系统信息-组件里查看NVCUDA.DLL的版本,两者是一样的

  2. 下载并安装cuda
    然后进入英伟达cuda下载网址 https://developer.nvidia.com/cuda-toolkit-archive 寻找对应版本的cuda进行下载。注意下载的是cuda toolkit,这是一个整合的工具箱。下载完之后进行傻瓜式
    安装,然后再次打开命令行,运行 nvcc -V进行检查是否安装完成。安装成功的话会显示如下的内容:

  • Pytorch安装
    pytorch框架是近年常用的框架之一,相比较于tensorflow等旧框架越来越来受到趋势的追捧。最新版的pytorch下载地址为:
    https://pytorch.org/get-started/locally/

pytorch的安装必须要根据cuda的版本来,有时候最新版本的pytorch还不是我们需要的。可以点击旁边的previous pytorch version查看先前版本的下载地址。由于pytorch支持cpu版本,因此要注意不要点错了。例如我的cuda版本是12.1,windows操作系统,如果想下载2.1版本的pytorch,应该要选择这个

其中的-c选项用于指定下载来源,如果有换源需求,可以把-c去掉,就会使用conda的默认下载源去下载。

最后,在下载完成之后,进入python,import torch,再使用torch.cuda.is_available(),如果返回True即表示成功。

标签:Pytorch,pytorch,CUDA,版本,显卡,安装,下载,cuda
From: https://www.cnblogs.com/maninfirer/p/18106663

相关文章

  • 从安装python开始教你利用python将excel中的数据导出到txt文件中,并且有如何安装python
    制作方法想到制作这个程序的原因开始写程序前的准备工作安装python第一步:下载python第二步:运行python安装包测试python用python输出你好python安装python库开始编写程序编写前的准备开始编写想到制作这个程序的原因工作的时候,领导说让把表格里的数据导出到txt文......
  • Docker 安装 GeoServer
    使用Docker安装GeoServer是一个更为便捷和灵活的方式,以下是在Linux上通过Docker安装GeoServer的简要步骤:安装Docker:如果您的系统还没有安装Docker,可以按照Docker官方文档(https://docs.docker.com/get-docker/)的指引进行安装。拉取GeoServer镜像:使用以下......
  • 文件系统 FTP Ubuntu 安装入门介绍
    FTP环境:Ubuntu14.04blogzh_CNubuntu14.04Install全新安装:apt-getinstallvsftpd重新安装:apt-get--reinstallinstallvsftpd卸载并清除配置文件:apt-get--purgeremovevsftpdStart&Restart$servicevsftpdstart$servicevsftpdrestart注意......
  • 安装并使用 Ubuntu Server 的一些注意事项
    最近,刚安装UbuntuServer22.04.4LTS版本,遇到些问题,记录其解决方案以备忘。 1)断网安装UbuntuServer系统采用USB启动盘,安装UbuntuServer时,一定要断网,因为连接互联网,安装UbuntuServer系统时,会自动下载更新,等待时间很可能会比较长,而断网会秒级安装,之后根据提示,重启......
  • 安装RabbitMQ
    安装方式安装RabbitMQ,官方提供很多种方式,具体参考地址如下https://www.rabbitmq.com/docs/download#installation-guidesRPM安装二进制下载安装源码构建安装DockerKubernetesVMware云厂商平台RPM安装TherearetwowaysofinstallingtheseRPMs:Installin......
  • cuda 流
    如下图,将多个执行相同核函数的进程通过cuda流来使他们并发执行,提升效率这很像cpu的流水线想让下面这个核函数执行两次,每次都是不同的参数我们需要用到cuda的流来并发的执行提升效率__global__voidkernel(int*a,int*b,int*c){intidx=threadIdx.x+blockId......
  • DataGrip的安装和破解
    DataGrip的安装和破解安装包获取方式:关注公众号爬虫探索者,发送datagrip即可领取。解压安装包下一步确定安装路径配置安装项菜单目录默认即可安装结束破解教程可以参考:pycharm破解安装......
  • Pycharm的安装和破解
    Pycharm的新版安装和破解1.解压文件无偿分享,只求关注个公众号:爬虫探索者,发送pycharm获取。2.双击安装2.1.点击下一步2.2.根据情况选择安装路径最好不要使用中文。2.3.勾选安装配置项只选择创建桌面快捷方式就行了,其他选项可以不用。2.4.开始菜单分类默认即可!......
  • 从零实战本地服务器部署 Docker 安装 ThingsBoard PE 专业版(适用于Cassandra + Kafka
    目录1、准备工作2、本地服务器LinuxCentos7.9系统安装docker2.1、检查Linux的内核版本2.2、卸载Docker旧版本(若有需要)2.3、安装Docker2.4、安装Docker引擎2.5、 启动docker和设置开机⾃启动3、使用Docker安装ThingsBoardPE3.1、 拉取ThingsBoardPE镜像3.2......
  • ubuntu编译与安装 OpenSSL-1.0.0
    apt-getpurgeopensslrm-rf/etc/ssl#删除配置文件编译与安装OpenSSLprefix是安装目录,openssldir是配置文件目录,另外建议安装两次,shared作用是生成动态连接库。(需要同时指定prefix与openssldir,否则可能会因为找不到文件而报错)wgetftp://ftp.openssl.org/source/op......