使用本教程前,默认您已经安装并配置好了python3以上版本
1. 去官网下载匹配的Cuda
Cuda下载地址
当前最高版本的Cuda是12.1
我安装的就是这个版本
小提示:自定义安装可以只选择安装Cuda Runtime。Nvidia全家桶不必全部安装。把全家桶全部安装完直接系统盘占了6G,很大的。
2. 安装pytorch
我之前使用的是pip install torch (2.0.0版本)
这样安装的torch是直接运行在CPU上的
想要使用GPU版本需要使用对应的cuda版本
尽管pytorch官网上目前只给出了11.8的Cuda支持,但是社区明确表明了兼容高版本Cuda。