win10下安装pytorch 1.8

最新推荐文章于 2024-06-29 12:16:29 发布

SOTVME

最新推荐文章于 2024-06-29 12:16:29 发布

阅读量600

点赞数

CC 4.0 BY-SA版权

文章标签： python

本文链接：https://blog.youkuaiyun.com/SOT3513909404/article/details/117964154

在Windows 10环境下配置PyTorch 1.8，包括检查NVIDIA显卡驱动，从PyTorch官网获取适配硬件的安装指令，使用conda进行安装，以及验证安装成功和安装PyG的详细步骤。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

##pytorch安装

win 10下安装pytorch

环境配置

环境配置

1.使⽤ nvidia-smi 命令查询显卡驱动是否正确安装
在这里插入图片描述
如果没有英伟达显卡，建议试试阿里云天池的DSW，新人有60小时免费的GPU运算时间。

2.进入官网查询pytorch，根据自己的硬件寻找相关安装语句。官网链接: pytorch官网.
在这里插入图片描述
选择自己的配置安装
接下来就耐心等待，下载内容较大。
也可以使用国内镜像安装，先设置安装源

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge 
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes

然后conda install pytorch