确认CUDA环境已经正确安装并且已经添加到PATH环境变量中。

确认CMake已经正确安装并且已经添加到PATH环境变量中。

确认NVIDIA® GPU驱动程序已经正确安装。

确认CUDA Toolkit已经正确安装。CUDA Toolkit可以在NVIDIA的官方网站下载到，下载地址为 https://developer.nvidia.com/cuda-toolkit 。

打开终端，输入以下命令来查看CMake是否能够找到CUDA编译器：

cmake --version

如果终端输出的信息中包含了CUDA_COMPILER，则表示CUDA编译器已经被正确安装。如果没有输出CUDA_COMPILER，则需要手动配置CUDA编译器路径。

打开终端，输入以下命令来获取CUDA编译器的路径：

which nvcc

终端会输出命令的绝对路径，例如 /usr/local/cuda-11.2/bin/nvcc。

打开终端，输入以下命令来手动配置CUDA编译器路径：

export CMAKE_CUDA_COMPILER=

将上述命令中的替换为第6步中获取到的CUDA编译器路径。

重新开始PyTorch的安装过程，应该就可以顺利安装了。

pip3 install torch torchvision torchaudio