pytorch-gpu 极简安装

1、进入pytoch官网：PyTorch

找到pytorch-gpu版本，看到CUDA11.8、11.7、CPU，这里我选择安装CUDA11.8

2、下载CUDA Toolkit：CUDA Toolkit 11.8 Downloads | NVIDIA Developer

3、下载CUDANN：cuDNN Download | NVIDIA Developer

命令行输入nvidia-smi，查看驱动信息

输入nvcc --version 进行检查

4、根据pytoch官方指令下载pytorch

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

5、验证：

>>> import torch
>>> torch.__version__
'2.0.0+cu118'
>>> torch.cuda.is_available()
True
>>> torch.cuda.device(0)
<torch.cuda.device object at 0x000001F2F10AAC70>
>>> torch.cuda.device_count()
1
>>> torch.cuda.get_device_name(0)
'NVIDIA GeForce GTX 1060 with Max-Q Design'

@著作：机智的小神仙儿

@时间：2023.7.31