PyTorch版本选择,安装,验证。

PyTorch 选择版本

打开Pytorch官网,

htps://pytorch.org/get-started/locally/Start Locallyhttp://htps://pytorch.org/get-started/locally/

根据需要选择要安装的PyTorch版本

在本例子中,这里选择最新的版本12.1生成的安装命令

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

注意,生成的安装指令默认是使用pip3,如果你用pip的改一下即可。

PyTorch 安装

PyTorch安装,开始执行安装,如下图:

提示安装完成,如下图:

验证PyTorch安装

验证PyTorch安装,执行下面python代码:

python 复制代码
import torch


if __name__ == '__main__':
    # 检测cuda环境
    print(torch.cuda.is_available())

验证结果为True,安装成功。

相关推荐
jndingxin41 分钟前
OpenCV 图像哈希类cv::img_hash::AverageHash
人工智能·opencv·哈希算法
ydl112843 分钟前
机器学习基础知识【 激活函数、损失函数、优化器、 正则化、调度器、指标函数】
python·机器学习
Jamence1 小时前
多模态大语言模型arxiv论文略读(153)
论文阅读·人工智能·语言模型·自然语言处理·论文笔记
晨曦5432101 小时前
量子计算突破:8比特扩散模型实现指数级加速
人工智能
Albert_Lsk1 小时前
【2025/07/11】GitHub 今日热门项目
人工智能·开源·github·开源协议
莫彩1 小时前
【大模型推理论文阅读】Enhancing Latent Computation in Transformerswith Latent Tokens
论文阅读·人工智能·语言模型
康斯坦丁师傅1 小时前
全球最强模型Grok4,国内已可免费使用!(附教程)
人工智能·grok
崔高杰1 小时前
微调性能赶不上提示工程怎么办?Can Gradient Descent Simulate Prompting?——论文阅读笔记
论文阅读·人工智能·笔记·语言模型
chao_7891 小时前
CSS表达式——下篇【selenium】
css·python·selenium·算法
元气小嘉1 小时前
前端技术小结
开发语言·前端·javascript·vue.js·人工智能