动手学深度学习配置

目录

安装pytorch-cpu

安装pytorch-gpu

---

 

安装pytorch-cpu

参考引用http://t.csdn.cn/3b3Hl

安装pytorch-gpu

参考http://t.csdn.cn/cfCzc

链接中的操作无法成功，参考其中的链接和具体安装图片即可，操作流程请看下文

以下操作中官网链接和少量细节请看参考链接

零 前期准备与CPU版本相同，python版本低于3.11就行，最好是3.9，就最后的安装语句不同，

一 确定电脑的显卡(英伟达)支持的最高CUDA版本，

二 然后去pytorch官网查看支持的最高的CUDA版本，

三 去CUDA官网下载对应稍微低于这两个最高版本的CUDA版本(最新的和最高的经常出问题)

四 在官网现在对应CUDA版本的cudnn版本，复制三个文件夹到CUDA

五 在pytorch官网复制对应版本安装代码(不要复制conda安装的代码，复制pip安装的代码，国内 conda下载有问题)

六  在prompt中复制下载代码，进行下载

七 安装 jupyter和d2l

pip install jupyter d2l

八 最后进行验证

在prompt输入

python

import pytorch

print(torch.cuda.is_available())