windows pytorch安装 cuda10.2
时间: 2023-09-26 14:05:44 浏览: 164
windows版本cuda10.2和cudnn以及cudart64_101.rar
您可以按照以下步骤在 Windows 上安装 PyTorch 和 CUDA 10.2:
1. 安装 Anaconda:如果您还没有安装 Anaconda,请下载并安装适用于 Windows 的 Anaconda。
2. 创建新的 Anaconda 环境:打开 Anaconda Prompt,输入以下命令创建新的环境。
```
conda create --name pytorch_env
```
3. 激活环境:输入以下命令激活新创建的环境。
```
conda activate pytorch_env
```
4. 安装 PyTorch:输入以下命令安装适用于 CUDA 10.2 的 PyTorch。
```
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
```
5. 验证安装:输入以下命令验证 PyTorch 是否安装成功。
```
python -c "import torch; print(torch.__version__)"
```
如果您在安装过程中遇到任何问题,请参考 PyTorch 官方文档以获得更多帮助。
阅读全文