annaconda配置pytorch gpu linux

### 配置Anaconda以使用PyTorch的GPU支持

#### 安装Anaconda
为了在Linux环境中配置PyTorch的GPU支持，首先需要安装Anaconda。可以通过wget下载Anaconda脚本并执行安装：

wget https://repo.anaconda.com/archive/Anaconda3-2024.10-1-Linux-x86_64.sh
bash Anaconda3-2024.10-1-Linux-x86_64.sh

完成安装后，建议重启终端使更改生效[^4]。

#### 初始化Anaconda环境
安装完成后可以选择是否初始化Anaconda。如果不希望每次启动shell都自动激活base环境，则可以在安装过程中跳过此选项，在需要时手动激活特定环境[^2]。

#### 创建新的Conda环境
创建一个新的Conda环境来隔离不同项目的依赖项是一个良好的实践方法。可以指定Python版本和其他必要的包一起安装：

conda create --name pytorch_env python=3.9

这会建立名为`pytorch_env`的新环境，并设定Python版本为3.9。

#### 查看CUDA版本
确保本地已经正确安装了CUDA工具链。可通过以下命令验证当前系统的CUDA版本：

nvidia-smi

该命令将显示有关NVIDIA驱动程序以及所安装的CUDA Toolkit的信息[^3]。

#### 安装带有CUDA支持的PyTorch
根据之前查到的CUDA版本选择合适的PyTorch版本进行安装。例如，对于CUDA 11.7，可使用如下命令：

conda install pytorch torchvision torchaudio cudatoolkit=11.7 -c pytorch

这条指令会在新创建的环境中安装与CUDA兼容的PyTorch及其相关库[^1]。

#### 激活虚拟环境
最后一步是激活刚刚创建好的环境以便开始工作：

conda activate pytorch_env

此时应该能够看到提示符前出现了`(pytorch_env)`字样，表示现在处于这个自定义环境中[^5]。