最简单的一文安装Pytorch+CUDA

1、不改清华源的前提下,官网下载太慢,如果换清华源又不能兼顾CUDA,最简单的办法直接多线程下载。小白可以用迅雷下载。

2、cuda版本和显卡支持版本需要自己看一下。

安装环境:window10
显卡:3050ti
使用Anaconda3-2024.06-1-Windows-x86_64
内置python版本是:python3.12.4
Pytorch版本:2.4.1+cu124
CUDA版本:12.4

1、安装版本及查看

没显卡的可以直接在Anaconda中的Anaconda Prompt

在命令提示行中直接输入

python 复制代码
pip install pytorch -i https://mirrors.tuna.tsinghua.edu.cn/pypi/web/simple

有显卡打开NVIDIA控制面板。

1、打开NVIDIA控制面板。

2、点击系统信息。

3、点击组件。查看NVIDIA CUDA 版本。

如图:我安装的是 12.4

这里建议的CUDA版本是11.8、12.1、12.4

如果CUDA版本是11.8-12.4范围内的可以跳过CUDA更新及安装。

建议去torch查看支持范围:Pytorch官网

确定好需要安装的CUDA版本后,访问NVIDIA的CUDA下载界面

安装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

下载太慢,直接复制链接通过迅雷下载

迅雷下载后,打开文件夹位置

在文件导航栏中输入CMD

点击回车并输入以下命令:

python 复制代码
pip install 

接着将下载的Pytorch拉倒CMD

点击回车运行。等待安装成功。

python 复制代码
python
import torch
torch.cuda.is_available()

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

相关推荐
小龙在山东1 分钟前
Python 包管理工具 uv
windows·python·uv
T__TIII5 分钟前
Dify 插件非正式打包
人工智能
jerwey9 分钟前
大语言模型(LLM)按架构分类
人工智能·语言模型·分类
IT古董12 分钟前
【第三章:神经网络原理详解与Pytorch入门】02.深度学习框架PyTorch入门-(5)PyTorch 实战——使用 RNN 进行人名分类
pytorch·深度学习·神经网络
令狐少侠201119 分钟前
ai之RAG本地知识库--基于OCR和文本解析器的新一代RAG引擎:RAGFlow 认识和源码剖析
人工智能·ai
weixin_3077791322 分钟前
批量OCR的GitHub项目
python·github·ocr
小叮当爱咖啡24 分钟前
Seq2seq+Attention 机器翻译
人工智能·自然语言处理·机器翻译
shadowcz00724 分钟前
奥特曼论人工智能、OpenAI与创业
人工智能·百度
AI人工智能+36 分钟前
应用俄文OCR技术,为跨语言交流与数字化管理提供更强大的支持
人工智能·ocr·文字识别
UQI-LIUWJ1 小时前
李宏毅LLM笔记: AI Agent
人工智能·笔记