12.1的cuda适配pytorch
时间: 2023-09-23 19:06:57 浏览: 1244
NVIDIA驱动、CUDA和Pytorch及其依赖
要在CUDA 12.1上使用PyTorch,您需要安装与CUDA 12.1兼容的PyTorch版本。您可以从PyTorch官方网站上下载适用于CUDA 12.1的预构建版本。或者,您也可以从源代码构建PyTorch,并在构建过程中指定CUDA 12.1。
以下是使用conda安装适用于CUDA 12.1的PyTorch的示例命令:
```bash
conda install pytorch torchvision torchaudio cudatoolkit=12.1 -c pytorch -c nvidia
```
请注意,您需要确保CUDA 12.1已正确安装并配置在您的系统上。此外,您的GPU也必须支持CUDA 12.1。
阅读全文