安装pytorch-cuda
时间: 2023-11-10 13:04:02 浏览: 72
mish-cuda:PyTorch 的 Mish 激活函数
安装 PyTorch-CUDA 可以通过以下步骤完成:
1. 确认你的电脑上已经安装了 CUDA,可以通过在命令行中输入 `nvcc --version` 来检查。
2. 打开 PyTorch 官网(https://pytorch.org/),选择相应的 PyTorch 版本和 CUDA 版本,复制对应的安装命令。
3. 在命令行中粘贴安装命令并运行,等待安装完成即可。
阅读全文