Ubuntu PyTorch与Anaconda如何搭配使用

以下是在Ubuntu上使用Anaconda搭配PyTorch的方法：

1. 安装Anaconda：从官网下载安装包，通过终端命令bash Anaconda3-xxx-Linux-x86_64.sh安装，安装过程中可按需选择安装路径等选项，安装完成后需将Anaconda添加到系统环境变量，可通过编辑~/.bashrc文件添加export PATH="/anaconda3/bin:$PATH"，然后执行source ~/.bashrc使配置生效。
2. 创建虚拟环境：使用conda create -n pytorch_env python=3.x命令创建，其中pytorch_env是环境名称，3.x是Python版本号，可根据需要选择。
3. 激活虚拟环境：通过conda activate pytorch_env命令激活创建好的虚拟环境。
4. 安装PyTorch：激活环境后，访问PyTorch官网，根据系统、CUDA版本等信息选择合适的安装命令，例如conda install pytorch torchvision torchaudio cudatoolkit=11.8 -c pytorch -c nvidia，若不需要CUDA则去掉cudatoolkit相关参数。
5. 验证安装：在激活的环境中，通过python进入Python交互环境，输入import torch，再输入print(torch.__version__)和print(torch.cuda.is_available())，若显示版本号且cuda可用则为安装成功。