pytorch安装教程

news/2024/10/23 7:36:45/

写在前面:配置pytorch着实有很多坑,不过最终结果算好的,话不多说,直接上干货。其中想要知道如何解决torch.cuda.is_available()返回false的,直接跳到步骤5pytorch安装。python版本至少是3.6及以上。

1、前言

        安装pytorch需要安装Anaconda、cuda、cuDnn、torch,本次机器配置如下:win10 64位操作系统,GTX3070,其中安装的上述版本如下(时间2023.9.24):

Anaconda3-5.3.1-Windows-x86_64  (去官网下载Download Success | Anaconda)

cuda_11.7.0_516.01_windows.exe(去官网下载即可CUDA Toolkit Archive | NVIDIA Developer 解释下为什么下载11.7,原因是cudcnn官网上只有11.7版本的,但是后续发现不一定要下载11.7版本,可以根据自己显卡来下载)

cudnn-windows-x86_64-8.7.0.84_cuda11-archive(该版本适用于11.XX版本的cuda,下载该版本即可,官网链接:Log in | NVIDIA Developer)

2、 Anaconda3安装

2.1 安装过程

下载链接:Download Success | Anaconda      

下载下来是一个Anaconda3-5.3.1-Windows-x86_64.exe

双击Anaconda3-5.3.1-Windows-x86_64.exe,全部直接默认即可。

安装完成后,在系统变量中添加环境变量,如下:

2.2 检查安装配置是否成功

Win+R输入cmd打开命令行 ,输入conda会显示相关命令选项

输入conda -V或者conda --version命令检查下安装配置是否成功

也可以输入conda info显示相关信息:

输入python会发现此时python也是带conda的版本

3、cuda安装

3.1 安装过程

下载链接:CUDA Toolkit Archive | NVIDIA Developer

本次选择cuda11.7版本进行安装,双击安装包默认即可完成安装。

3.2 检查安装是否成功

Win+R输入cmd打开命令行 ,输入 nvcc --version,可以看到我安装的是11.7

4、安装cuDNN

4.1 环境配置

安装cuDNN版本根据cuda进行选择(下载链接cuDNN Archive | NVIDIA Developer),由于本文是cuda11.7,因此选择了cuDNN8.7版本。

下载的cuDNN是一个压缩包,将其解压缩后,得到如下文件:

此时,将bin、include、lib文件复制粘贴到cuda安装目录下,里下图:

4.2 检查安装是否成功

Win+R输入cmd打开命令行 ,输入 cd C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.7\extras\demo_suite (该路径根据自己安装实际路径,进入该路径下,运行bandwidthTest.exe与deviceQuery.exe,如果运行结果result均为pass,则表明安装成功。

5、 pytorch安装

这也是本博客与很多博客不同的地方,如果按照很多博客方法从官网下,使用pip或者conda安装torch均错误,在torch.cuda.is_available()返回false,即torch无法调用GPU。

这是由于在国内不好下载GPU-cuda,只有CPU-cuda。下面给出一种切实可行的方案。

去网站https://download.pytorch.org/whl/torch/,进入torch下载对应的带GPU的torch

进到torch,可以看到有各种版本的torch的whl文件,如:torch-1.13.0+cu116-cp37-cp37m-win_amd64.whl,意为torch1.13.0、cuda11.6、python3.7、win 64位系统的版本。

找到你电脑上安装的python、cuda版本,比如我的电脑是python=3.7、cuda=11.7(另外说一下,python的版本是你从cmd控制台上查询的版本,而不是你电脑上安装的版本!!!)所以,最后我选择的版本如下

(PS:在安装python虚拟环境时,python版本最高大于等于3.7,因为在上述网站中没有cp36的版本。)

    选好whl文件后,下到本地,然后打开anaconda powershell prompt(anaconda3) (cmd应该也行),切换到自己的虚拟环境下。

如果你没有创建自己的虚拟环境,其创建过程如下,打开anaconda prompt,输入如下命令,其中python版本是你电脑上安装的版本

conda create -n pytorch python=3.7

PS:倘若无法激活自己创建的虚拟环境torch,采用如下命令:

先输入conda info --envs

再输入 activate

最后输入 conda activate pytorch

在创建好的虚拟环境后,再运行命令:

pip cache purge 

再运行安装之前下载的whl文件即可,如下:

pip3 install E:\soft\major\Pytorch\torch-1.13.1+cu117-cp37-cp37m-win_amd64.whl

稍等会,就安装好了。

如果重新进入,点击anaconda prompt,输入 conda info --envs

再输入conda activate pytorch

再输入python

再输入 import torch

最后输入 torch.cuda.is_available()

结果返回true,

后面就可以使用cuda干活了


http://www.ppmy.cn/news/1123308.html

相关文章

[数据结构】栈和队列

目录 1.栈 1.1概念 1.2 栈的使用 1.3.栈的模拟实现 2.队列 2.1概念 2.2队列的使用 2.3队列的模拟实现 2.4 循环队列 2.5双端队列 1.栈 1.1概念 栈:一种特殊的线性表,其只允许在固定的一端进行插入和删除元素操作。进行数据插入和删除操作的一…

实验室安全教育与考试

目录 我的错题(2个)新知识题目(10个)刚开始不太理解的题目(10个)写在最后(免责声明) 我的错题(2个) 18.发生电气火灾时可以使用的灭火设备包括:&…

MySQL索引事务

一 、 索引 索引是一种特殊的文件,包含着对数据表里所有记录的引用指针。 可以对表中的一列或多列创建索引, 并指定索引的类型,各类索引有各自的数据结构实现。 索引保存的数据结构主要为B树,及hash的方式。 1. 作用 数据库中…

中秋国庆内卷之我爱学习C++

文章目录 前言Ⅰ. 内联函数0x00 内联函数和宏的比较0x01 内联函数的概念0x02 内联函数的特性 Ⅱ. auto(C 11)0x00 auto的概念0x01 auto的用途 Ⅲ. 范围for循环(C11)0x00 基本用法0x01 范围for循环(C11)的使用条件 Ⅳ. 指针空值nullptr(C11)0x00 概念 前言 亲爱的夏…

gitee生成公钥和远程仓库与本地仓库使用验证

参考文档: https://help.gitee.com/base/account/SSH%E5%85%AC%E9%92%A5%E8%AE%BE%E7%BD%AE(1)通过命令ssh-keygen 生成SSH key -t key类型 -c注释 ssh-keygen -t ed25519 -C "Gitee SSH Key" (2)按三次回车 (3)查看生成的 SSH 公钥和私钥: …

新增MariaDB数据库管理、支持多版本MySQL数据库共存,1Panel开源面板v1.6.0发布

2023年9月18日,现代化、开源的Linux服务器运维管理面板1Panel正式发布v1.6.0版本。 在这个版本中,1Panel新增MariaDB数据库管理;支持多版本MySQL数据库共存;支持定时备份系统快照和应用商店中已安装应用;支持为防火墙…

elementui实现表格(el-table)默认选中

方法: 创建一个空数组用来存放默认数据 遍历表格的数据,再遍历需要在表格中反显的数据,两者的id一致(两者之间共同的标识即可,一般以id做判断) 把判断出来的默认表格数据push到创建的数组中 再遍历数组,将数…

5、SpringBoot_热部署

六、热部署 1.热部署概述 概述&#xff1a;程序更改后&#xff0c;不需要重新启动服务器也能够实现动态更新 springboot 项目如何实现热部署&#xff1f; tomcat 已经内置到项目容器中了希望tomcat监听外部程序变化通过新建一个程序来监控你代码的变化 2.依赖导入 依赖 <…