Ubuntu1804搭建Pytorch从Anaconda开始

1 篇文章 0 订阅
  • 2
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
要在 Ubuntu 20.04 上搭建 PyTorch 3D 模型训练环境,你可以按照以下步骤进行操作: 1. 安装 CUDA 和 cuDNN(如果你的 GPU 支持 CUDA): - 首先,检查你的 GPU 支持的 CUDA 版本,并从 NVIDIA 官方网站下载对应版本的 CUDA Toolkit。 - 安装 CUDA Toolkit,按照官方文档提供的说明进行安装。 - 在安装 CUDA Toolkit 后,下载对应版本的 cuDNN,并按照官方文档提供的说明进行安装。 2. 安装 Anaconda(可选): - 访问 Anaconda 官方网站并下载适用于 Linux 的 Anaconda 安装包。 - 打开终端,导航到下载的 Anaconda 安装包所在的目录,并执行以下命令安装 Anaconda: ``` bash Anaconda3-2021.05-Linux-x86_64.sh ``` - 按照安装程序的提示进行安装。 3. 创建并激活虚拟环境: - 打开终端,执行以下命令创建一个名为 "pytorch" 的虚拟环境: ``` conda create -n pytorch python=3.8 ``` - 激活虚拟环境: ``` conda activate pytorch ``` 4. 安装 PyTorch 和 torchvision: - 在激活的虚拟环境中,执行以下命令安装 PyTorch 和 torchvision: ``` conda install pytorch torchvision torchaudio cudatoolkit=xx.x -c pytorch ``` (请使用你安装的 CUDA 版本的正确替代项 xx.x) 5. 安装其他依赖项: - 执行以下命令安装其他常用的 PyTorch 相关依赖项: ``` conda install matplotlib numpy pandas ``` 这样,你就可以在 Ubuntu 20.04 上成功搭建 PyTorch 3D 模型训练环境了。记得在使用 PyTorch 时,始终在激活的虚拟环境中运行代码。
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值