最简单的一文安装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就代表可以安装完成了。

相关推荐
算家计算3 分钟前
AI大神100美元手搓ChatGPT!nanochat教程爆火,4小时炼成聊天机器人
人工智能·chatgpt·资讯
Kaydeon4 分钟前
【AIGC】50倍加速!NVIDIA蒸馏算法rCM:分数正则化连续时间一致性模型的大规模扩散蒸馏
人工智能·pytorch·python·深度学习·计算机视觉·aigc
阿里云大数据AI技术18 分钟前
AI刷新赛事体验,PAI-ArtLab支撑“我的NBA AI手办”互动
人工智能
B站_计算机毕业设计之家25 分钟前
大数据实战:Python+Flask 汽车数据分析可视化系统(爬虫+线性回归预测+推荐 源码+文档)✅
大数据·python·数据分析·flask·汽车·线性回归·预测
晚枫~27 分钟前
零基础快速上手Playwright自动化测试
javascript·python·测试工具·c#·自动化
zhy2956330 分钟前
【DNN】基础环境搭建
人工智能·tensorrt·cuda·开发环境·cudnn
PKNLP32 分钟前
NLP入门
人工智能·自然语言处理
Mr.Lee jack34 分钟前
体验GPT-OSS-120B:在PH8平台上探索超大规模语言模型的强大能力
人工智能·gpt·语言模型·openai·deepseek
余衫马35 分钟前
大语言模型(LLM)领域细分方向解析
人工智能·语言模型·自然语言处理·llm·领域方向
小关会打代码37 分钟前
自然语言处理(NLP)之分词
人工智能·自然语言处理