win 10 pytorch cuda

Windows 10 上安装 PyTorch 并启用 CUDA（Compute Unified Device Architecture）通常涉及几个步骤：

1. **硬件要求**：首先，你需要一台支持CUDA的NVIDIA GPU，并且需要满足最低的CUDA版本要求。检查你的GPU是否兼容及驱动程序是否更新。

2. **安装CUDA**：访问NVIDIA官网下载适用于Windows 10的CUDA工具包，包括CUDA Toolkit、Driver和CUDAamples。按照提示进行安装，确保在安装过程中选择“Include CUDAamples”选项以便测试。

3. **安装Python和PyTorch**：推荐使用Anaconda或Miniconda创建一个新的虚拟环境，然后通过pip安装Python和PyTorch。命令示例如下：

   conda create -n pytorch python=3.8
   conda activate pytorch
   pip install torch torchvision cudatoolkit=11.0

这里假设你想要使用CUDA 11.0，记得替换为你实际下载的CUDA版本。

4. **验证安装**：安装完成后，可以运行一些例子测试CUDA是否正常工作，如`torch.cuda.is_available()`会返回True如果成功。

5. **设置路径**：确保系统环境变量中包含CUDA的bin目录，这样才能让Python在运行时找到CUDA库。

6. **添加CUDNN**：如果你的应用涉及到深度学习，可能会需要安装CuDNN（CUDA Deep Neural Network Library）。同样是从NVIDIA官网获取并安装对应版本。