windows/ubuntu/Pytorch安装
windows:
Ubuntu:
ubutu-drivers devices
输出显卡信息,找recommended驱动信息
然后到software&update-additional drivers找到对应的驱动,Apply
提示restart,开机后测试
nvidia-smi
我的是这样就输出成功了,也可参考其他博主安装方式
Pytorch
PyTorch:Pytorch,cuda12.2可安装11.8向下兼容
直接在这个网址上下载自己需要的安装包,本地安装
conda/pip
pip临时使用清华源:pip install -i https://pypi.tuna.tsinghua.edu.cn/simple