当前位置: 首页 > article >正文

第一次安装Pytorch

1、新版本的Anaconda内置的python版本是3.12,
目前 Windows 上的 PyTorch 仅支持 Python 3.8-3.11;不支持 Python 2.x。

在这里插入图片描述

1、创建运行环境

在不创建虚拟环境的情况下,不建议使用最新的Python和Anaconda。
在几次失败后,我使用的是Anaconda3-2023.09-0-Windows-x86_64版本。
在清华源下载:Anaconda3-2023.09-0-Windows-x86_64
在这里插入图片描述
因为下载的是旧版Anaconda,很多自带库都需要更新。不过这都是非必要的。安装Anaconda教程可以参考 **一文安装anaconda **

安装好Anaconda,在开始菜单中找到 Anaconda Prompt,右键单击,点击更多打开文件位置,将 Jupyter Notebook和Anaconda Prompt复制到桌面上。
在这里插入图片描述
双击打开Anaconda Prompt,输入命令更新pip

python -m pip install --upgrade pip

2、安装cuda

可以先查看自己的cuda版本,
1、打开NVIDIA控制面板。
2、点击系统信息。
3、点击组件。查看NVIDIA CUDA 版本。
如图:我安装的是 12.4

在这里插入图片描述
这里建议的CUDA版本是11.8、12.1、12.4

安装CUDA方法:点击进入 CUDA-toolkit-archive

选择合适的版本进行安装。
譬如:11.8、12.1、12.4
在这里插入图片描述

3、安装Pytorch

这个是截止到2024年9月,Pytorch的安装环境需求。
在这里插入图片描述

如果CUDA版本同样是12.4,可以直接运行以下代码:

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124

如果CUDA版本是12.1,可以直接运行以下代码:

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

如果CUDA版本同样是11.8,可以直接运行以下代码:

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

打开Anaconda Prompt,输入上述对应版本的命令。
在这里插入图片描述
发现时间太长,可以更换pip源。
我想使用清华源下载,发现还是太慢。
最后使用迅雷下载whl,然后pip下载。
情况如下:
复制上述链接:
https://download.pytorch.org/whl/cu124/torch-2.4.1%2Bcu124-cp311-cp311-win_amd64.whl
打开迅雷,下载。
Anaconda Prompt中输入

pip install "C:\Users\程辉\Downloads\torch-2.4.1+cu124-cp311-cp311-win_amd64.whl" 

我下载的Pytorch是在
E:\迅雷下载\torch-2.4.1+cu124-cp311-cp311-win_amd64.whl
如果不想输入位置,可以先输入 pip install 后将文件夹中的torch拖入到Anaconda Prompt中,回车运行。
等待安装完成,会显示:torch is already installed…
可以尝试输入以下代码进行检测
在这里插入图片描述

python
import torch
torch.cuda.is_available()

返回True就代表可以安装完成了。


http://www.kler.cn/a/313758.html

相关文章:

  • Java的dto,和多表的调用
  • 使用jmeter查询项目数据库信息,保存至本地txt或excel文件1108
  • 轻松上手:使用Docker部署Java服务
  • Systemd: disable和mask的区别
  • 论软件维护及其应用子问题
  • Java基于SpringBoot+Vue的宠物共享平台的设计与实现(附源码,文档)
  • python爬虫初体验(二)
  • Android横竖屏 mdpi hdpi xhdpi xxhdpi xxxhdpi
  • .ideavimrc在idea打不开
  • 基于NLP的对话系统开发:从零构建智能客服机器人
  • Android 如何使用jdk命令给应用/APK重新签名。
  • 云栖大会Day1:云应用开发平台 CAP 来了
  • pythonnet python图像 C# .NET图像 互转
  • FLUX.1+ComfyUI部署与使用:图像合成技术的新高度
  • 【JAVA开源】基于Vue和SpringBoot的在线文档管理系统
  • oracle dblink的使用并举例
  • Spring AOP的应用
  • 【计算机网络 - 基础问题】每日 3 题(十三)
  • 面试八股--stram 中map和flatmap的区别
  • 关于 NLP 深度学习的核心流程
  • 虚拟现实与PD协议快充
  • thinkphp 做分布式服务+读写分离+分库分表(分区)(后续接着写)
  • 【网络安全的神秘世界】攻防环境搭建及漏洞原理学习
  • LeetCode 459.重复的子字符串
  • 计算机网络分类
  • Ubuntu 22.04 源码下载、编译