当前位置: 首页 > 后端技术 > Python

Manjaro下安装Pytorch注意事项

时间:2023-03-26 15:55:33 Python

1.在软件安装器中找到pytorch,我需要安装CUDA加速,选择下面的python-pytorch-cuda包:2.在manjaro设置中查看系统驱动版本,可以看到我的驱动是440xx:3.在安装管理器里的selection里选择你系统对应的显卡硬件驱动:4.安装好torch之后,记得安装torchvision:5.然后在终端输入python,执行下面命令,没有错误提示已经安装:Python3.8.1(default,Jan222020,06:38:00)[GCC9.2.0]onlinuxType"help","copyright","credits"or"许可证”以获取更多信息。导入torchimporttorchvision