本来是要在有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复制。
大家使用的如果是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粘贴进去回车,即可静等安装。
完成安装后在同一界面输入ipython,打开命令行ipython编辑器,导入pytorch包测试是否成功。
import torch as t
如果没有报错的话,就安装成功了。
GPU安装
主要是注意CUDA版本的选择。
因为GPU需要先安装CUDA和cudnn,所以需要在NVIDIA设置中点选“管理3D设置”-“系统设置”-“组件”,查看对应CUDA版本号。我们下载的版本号只能比它旧不能新。
之后步骤见文章。
CUDA、CUDNN在windows下的安装及配置_花花少年的博客-CSDN博客_cudnn配置
真的很详细,我也很想这么装,无奈笔记本身板受不住。