1.查看自己要安裝的cuda版本、對應(yīng)的driver烦磁。對應(yīng)的網(wǎng)址為:https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
如果版本不對應(yīng),可以先根據(jù)更新顯卡驅(qū)動到高版本,網(wǎng)址如下:https://www.nvidia.cn/Download/index.aspx?lang=cn
2.下載網(wǎng)址:https://developer.nvidia.com/cuda-toolkit-archive亏钩。安裝cuda默認(rèn)安裝即可,和cudnn下載網(wǎng)址:https://developer.nvidia.com/rdp/cudnn-archive欺旧,二者一定那個要匹配姑丑。
3.下載torch是一定要版本對應(yīng),可以直接指定辞友。https://download.pytorch.org/whl/cu117