第一次安装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 复制代码
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.812.112.4

3、安装Pytorch

这个是截止到2024年9月,Pytorch的安装环境需求。

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

python 复制代码
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124

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

python 复制代码
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

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

python 复制代码
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中输入

python 复制代码
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 复制代码
python
import torch
torch.cuda.is_available()

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

相关推荐
理想是做个英雄14 分钟前
基于AC6366C做AI语音鼠标
人工智能·计算机外设·杰理ble,mcu以及音频蓝牙
stbomei29 分钟前
2025 AI 消费端变革:从生活助手到体验重构的全民浪潮
人工智能·重构·生活
琥珀食酒社37 分钟前
马云归来,“新零售”不死
人工智能·零售
IT_陈寒42 分钟前
JavaScript性能优化:7个90%开发者不知道的V8引擎黑科技
前端·人工智能·后端
tirvideo43 分钟前
RK3588芯片与板卡全面解析:旗舰级AIoT与边缘计算的核心
人工智能·嵌入式硬件·深度学习·目标检测·机器学习·计算机视觉·边缘计算
程序猿小D1 小时前
【完整源码+数据集+部署教程】【智慧工地监控】建筑工地设备分割系统: yolov8-seg-efficientViT
python·yolo·计算机视觉·数据集·yolov8·yolo11·建筑工地设备分割系统
努力也学不会java1 小时前
【Java并发】揭秘Lock体系 -- 深入理解ReentrantLock
java·开发语言·人工智能·python·机器学习·reentrantlock
DatGuy1 小时前
Week 18: 深度学习补遗:Stacking和量子运算Deutsch算法
人工智能·深度学习·算法
红苕稀饭6662 小时前
RISE论文阅读
论文阅读·人工智能·计算机视觉
cxr8283 小时前
分享一个知识工程师单体智能体的简单提示词
人工智能·智能体·提示词工程·ai赋能·上下文工程