如何安装torch cuda 11 - Shell/Bash

在开始之前，请确保您已经安装了以下软件：

• NVIDIA显卡驱动程序
• CUDA Toolkit
• cuDNN

如果您还没有安装，可以参考以下链接进行安装：

• NVIDIA显卡驱动程序：https://www.nvidia.com/en-us/drivers/
• CUDA Toolkit：https://developer.nvidia.com/cuda-toolkit-archive
• cuDNN：https://developer.nvidia.com/rdp/cudnn-download

步骤1：安装Anaconda

如果您已经安装了Anaconda，则可以跳过此步骤。如果尚未安装，请根据您的操作系统和系统架构下载并安装Anaconda：

https://www.anaconda.com/products/individual

步骤2：创建虚拟环境

使用以下命令创建一个新的虚拟环境，名称为“myenv”，python版本为3.8：

conda create -n myenv python=3.8

步骤3：激活虚拟环境

使用以下命令激活新创建的虚拟环境：

conda activate myenv

步骤4：安装PyTorch和TorchVision

使用以下命令安装PyTorch和TorchVision：

conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch -c nvidia

步骤5：验证安装

使用以下代码片段验证PyTorch和TorchVision是否已在您的系统上正确安装：

import torch
print(torch.__version__)
print(torch.cuda.is_available())

输出应该与以下内容类似：

1.9.0+cu111
True

恭喜您，现在您已经成功安装了PyTorch和TorchVision！