在 Linux 上轻松部署 PyTorch 环境:一步步指南
# 安装Anaconda
wget https://repo.anaconda.com/archive/Anaconda3-2020.11-Linux-x86_64.sh
sh Anaconda3-2020.11-Linux-x86_64.sh
# 安装PyTorch
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch -c conda-forge
# 验证PyTorch安装
python -c "import torch; print(torch.__version__)"
这段代码展示了如何在Linux系统上安装Anaconda,并利用它来安装PyTorch及其相关的库。代码中使用了wget来下载Anaconda安装脚本,并使用sh命令来运行它。然后使用conda命令来安装PyTorch及其依赖项,其中cudatoolkit的版本是11.3,来自pytorch和conda-forge两个channel。最后,使用Python的-c选项来验证PyTorch的版本。
评论已关闭