pytorch安装

最新推荐文章于 2025-02-13 22:36:57 发布

Hairy-

最新推荐文章于 2025-02-13 22:36:57 发布

阅读量1k

点赞数

分类专栏：深度学习文章标签： pytorch python

本文链接：https://blog.csdn.net/weixin_45321349/article/details/121487088

版权

深度学习专栏收录该内容

1 篇文章 0 订阅

订阅专栏

本文详细介绍了如何下载和安装PyTorch，包括使用Anaconda创建环境，检查CUDA版本，从官方页面下载对应版本的库文件，并通过pip在指定环境中安装PyTorch、torchvision和torchaudio。文章还提供了遇到下载速度慢时如何使用清华源加速的提示，以及安装完成后验证安装是否成功的步骤。

摘要由CSDN通过智能技术生成

第一步

下载anaconda（用迅雷下载）*

第二步

在桌面点击右键，选择“NVIDIA控制面板”
依次点击：帮助–>系统信息
点击“组件”，本电脑CUDA是11.4的

第三步

到 https://download.pytorch.org/whl/torch_stable.html 下载对应的三个文件(我下载到pkgs目录下)
技巧： 可以打开pytorch的官网PyTorch查看自己适合什么版本（注意选择CUDA版本！），然后再进入上面的网站单独下载。
注意选择pip安装

第四步

首先打开cmd
使用“conda activate xxxxx（之前创建的环境名字）”激活到当前环境
再使用cd定位到下载torch的目录中
最后依次pip install torch、torchvison、torchaudio（一定要输入完整的名字）
全部安装完了试一试输入“python”–>“import torch”。如果没有报错即安装成功。
附：安装过程中可能有些包的下载速度有些慢，可以使用清华源替换.condarc这个文件，详细参考清华源的界面介绍

原文链接：https://blog.csdn.net/JaysonRen/article/details/114918061

也可以在线安装

pip install torch1.4.0+cpu torchvision0.5.0+cpu -f https://download.pytorch.org/whl/torch_stable.html