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文章来源:https://www.toymoban.com/news/detail-621362.html
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
5、验证:文章来源地址https://www.toymoban.com/news/detail-621362.html
>>> 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
到了这里,关于pytorch-gpu 极简安装的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!