动手学深度学习笔记|1.安装带GPU的pytorch版本

来源: myluster的github笔记，求个star说是

环境安装指南

大体依照动手学深度学习书上进行安装,但不安装d2l库，并且展开说说Pytorch的安装问题

PyTorch GPU版本安装

• 请先安装与自己硬件驱动版本相应的CUDA，并获取自己的CUDA版本，具体过程不展开

安装对应的pytorch版本

# 如我的CUDA版本是11.8，则有
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
（我这里使用了conda安装）

验证安装是否成功

#在cmd中输入
python
import torch
print(torch.cuda.is_available())

• 输出true就正常了,如果输出false，建议首先看看cuda版本是否正确

不安装d2l库，而是直接安装以下库（待更新）

Matplotlib

注意事项

• 【1】由于本人使用原始人编译器jupyter notebook进行学习，如果使用时出现内核崩溃，请尝试在cmd中输入以下代码

pip install --upgrade jupyter jupyterlab numpy pandas matplotlib scikit-learn