PyTorch 1.3.1在Windows下载并且计算机配置了 NVIDIA GPU通过conda下载链接为多少
时间: 2024-02-21 21:57:00 浏览: 233
PyTorch1.3.1官方文档.zip
您可以通过以下链接下载PyTorch 1.3.1 GPU版本:
```
conda install pytorch=1.3.1 torchvision cudatoolkit=<version_number> -c pytorch
```
请将`<version_number>`替换为您所安装的CUDA版本号。例如,如果您的计算机安装了CUDA 10.1,则应该使用以下命令:
```
conda install pytorch=1.3.1 torchvision cudatoolkit=10.1 -c pytorch
```
注意,您需要在安装PyTorch之前安装NVIDIA驱动程序和CUDA工具包。另外,如果您的计算机还配备了cuDNN,建议您也一并安装以提高性能。
阅读全文