anaconda 安装pytorch-gpu
时间: 2023-10-05 16:11:03 浏览: 131
你可以使用 Anaconda 来安装 PyTorch-GPU。首先,确保你已经安装了最新版本的 Anaconda,并且已经正确安装了 NVIDIA GPU 驱动程序和 CUDA 工具包。
接下来,打开 Anaconda Prompt 或者终端,并创建一个新的虚拟环境(可选但推荐),可以使用以下命令:
```
conda create -n pytorch_gpu_env python=3.8
conda activate pytorch_gpu_env
```
然后,安装 PyTorch-GPU,运行以下命令:
```
conda install pytorch torchvision torchaudio cudatoolkit=<your_cuda_version> -c pytorch
```
请将 `<your_cuda_version>` 替换为你所安装的 CUDA 版本号(例如,`cu111`、`cu112`等)。如果你不确定你的 CUDA 版本,可以在命令行中运行 `nvcc --version` 来查看。
安装完成后,你就可以在你的 Python 代码中导入 PyTorch 并使用 GPU 加速了:
```python
import torch
# 检查是否使用了 GPU
print(torch.cuda.is_available())
# 在 GPU 上创建一个 Tensor
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
x = torch.tensor([1., 2.], device=device)
```
希望这对你有所帮助!如果有任何问题,请随时问我。
阅读全文