Pytorch+win10+cpu版本pip安装(内附gpu版本链接)

本来是要在有NVIDIA显卡的笔记本上搞事的,无奈C盘太小容不下CUDA+Pytorch这尊大神(GPU包直接3个G,CPU包一共 250MB都不到),所以就只能普通做一下记录。

此文前提是已安装python和anaconda。

CPU版

为确保下载方便,使用pip下载。

在Pytorch官网的以往版本中选择符合的版本下载(对cpu要求不高,不过这点对gpu一定要注意)

Previous PyTorch Versions | PyTorch

(我选的是v1.10.0)下滑到Wheel板块,找到CPU only下面这行代码,全部选中后ctrl+c复制。Pytorch+win10+cpu版本pip安装(内附gpu版本链接)_第1张图片

大家使用的如果是Python3的话,可以先复制到txt中,pip后可以加个3,再复制到cmd中运行下载。

pip3 install torch==1.10.0+cpu torchvision==0.11.0+cpu torchaudio==0.10.0 -f https://download.pytorch.org/whl/torch_stable.html

打开anaconda3的Anaconda Prompt,ctrl+v粘贴进去回车,即可静等安装。

Pytorch+win10+cpu版本pip安装(内附gpu版本链接)_第2张图片

 完成安装后在同一界面输入ipython,打开命令行ipython编辑器,导入pytorch包测试是否成功。

import torch as t

如果没有报错的话,就安装成功了。

Pytorch+win10+cpu版本pip安装(内附gpu版本链接)_第3张图片

GPU安装

主要是注意CUDA版本的选择。

因为GPU需要先安装CUDA和cudnn,所以需要在NVIDIA设置中点选“管理3D设置”-“系统设置”-“组件”,查看对应CUDA版本号。我们下载的版本号只能比它旧不能新。

Pytorch+win10+cpu版本pip安装(内附gpu版本链接)_第4张图片

 之后步骤见文章。

CUDA、CUDNN在windows下的安装及配置_花花少年的博客-CSDN博客_cudnn配置

 真的很详细,我也很想这么装,无奈笔记本身板受不住。

你可能感兴趣的:(pytorch)