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

第一次安装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 E:\迅雷下载\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.mrgr.cn/news/28654.html

相关文章:

  • 分布式光伏智慧平台建设现场 系统集成商如何盈利
  • Conda环境、Ubuntu环境移植
  • 如何快速定位并解决 Linux 系统性能瓶颈:终极全攻略
  • 数据库SQLite的使用
  • 论文阅读《机器人状态估计中的李群》
  • 【JavaWeb】JavaWeb入门之XML详解
  • MessagesPlaceholder
  • uniapp中实现<text>文本内容点击可复制或拨打电话
  • [性能]高速收发的TCP/MQTT通信
  • 微服务_1、入门
  • 使用卷积神经网络进行人类活动识别的特征学习:惯性测量单元和音频数据的案例研究
  • macOS Sequoia 15 发布,iPhone 镜像、密码应用程序、窗口平铺更新等带来全新体验
  • 云原生信息安全:筑牢数字化时代的安全防线
  • 107. 超快速排序
  • 系统架构设计师教程 第5章 5.7 软件项目管理 笔记
  • [Java]maven从入门到进阶
  • Linux基础---07文件传输及解决yum安装失效的方法
  • 年化60.7%,最大回撤-16.5%,RSRS标准分择时效果差不多
  • pytorch 模型训练太慢怎么办,试一试这17种方法可以优化训练过程,pytorch 提高训练速度的方法 除了num_worker
  • 【小白】一文安装anaconda
  • Java创建者模式(一)——单例设计模式(饿汉式、懒汉式、枚举式 | 完成详解,附有代码+案例)
  • docker容器镜像服务配置
  • Vue学习记录之一(介绍及脚手架的使用)
  • 深入解析 Cursor:AI 驱动的编程工具与应用示例
  • RabbitMQ Spring客户端使用
  • 道路裂缝,坑洼,病害数据集-包括无人机视角,摩托车视角,车辆视角覆盖道路