在anaconda中安装pytorch

最新推荐文章于 2024-08-08 20:15:38 发布

weixin_45965693

最新推荐文章于 2024-08-08 20:15:38 发布

阅读量1k

点赞数 1

分类专栏：深度学习

本文链接：https://blog.csdn.net/weixin_45965693/article/details/113796112

版权

深度学习专栏收录该内容

8 篇文章 0 订阅

订阅专栏

本文详细记录了在Windows10上安装PyTorch的完整步骤，包括检查CUDA版本、使用conda快速安装以及Jupyter Notebook的配置。特别提示，为提高下载速度，建议预先切换到清华源。

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

一些有必要搞清楚的概念
windows10安装pytorch，感觉自己基本踩了所有坑，这里提供最有效的两种方法
在这里插入图片描述

没有切换源的请跳到最后看STEP0

STEP1 查看cuda版本
桌面右键 ‘NAVIDA控制面板’
‘帮助’—‘系统信息’—‘组件’
在这里插入图片描述
于是，cudatoolkit=10.2

STEP2 安装
直接复制粘贴官网的命令好像不行
由于我太菜了在虚拟环境中安装时间长，也肯定会遇到问题又耗时了，
所以直接在base下安装，命令如下

conda install pytorch torchvision cudatoolkit=10.2 -c pytorch -c defaults -c numba/label/dev

可以仔细看看package plan，少了许多在这里插入图片描述
STEP3 Jupyter Notebook检查成功

STEP4 切换Jupyter Notebook文件夹

C:\Users\Administrator\AppData\Roaming\Microsoft\Windows\Start Menu\Programs\Anaconda3 (64-bit)

如果下载了代码可能用到，在起始位置改成你需要打开的文件夹
在这里插入图片描述

STEP0
注：如果下载速度不快，应该提前切换清华源（因为豆瓣源可能不全）
Anaconda切换清华源(最全的镜像切换)
CMD输入以下命令：

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/win-64/ 
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/win-64/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2/win-64/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/mro/win-64/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/pro/win-64/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r/win-64/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/numba/label/dev/win-64/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/win-64/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/win-64/
conda config --set show_channel_urls yes