PyTorch配置CUDA

硬件要求：

1. 显卡支持所需版本CUDA

2. 显卡驱动程序支持所需版本CUDA

3. PyTorch与显卡计算能力匹配

1. 检查GPU/显卡是否支持所需的 CUDA 版本

step1：确定显卡型号

windows右击选择计算机管理→显示适配器

注意：只有NVIDIA显卡支持CUDA，AMD以及Intel显卡不支持。

step2：查看显卡支持的CUDA版本（https://en.wikipedia.org/wiki/CUDA#GPUs_supported）

1. 找到显卡类型以及对应的算力（表格不全，可以自行搜索显卡类型对应算力）

2. 在表格前面的说明中找到算力对应的CUDA版本

1. 检查GPU/显卡驱动是否支持所需的 CUDA 版本

建议下载最新版本驱动

1. 下载CUDA对应的PyTorch

PyTorch

例如，我的CUDA版本是11.6，使用pip进行下载，打开页面提示网址https://download.pytorch.org/whl/cu116

选择torch，查找cu116，我的python版本是8.5，windows系统，所以选择进行下载torch-1.12.0+cu116-cp38-cp38-win_amd64.whl

下载完成后，将下载好的whl文件拖到环境中，利用

pip install torch-1.12.0+cu116-cp38-cp38-win_amd64.whl

进行安装。

安装完成后运行

注意：文章参考

https://stackoverflow.com/questions/60987997/why-torch-cuda-is-available-returns-false-even-after-installing-pytorch-with#comment125786629_60987997