pytorch环境搭建

以下是搭建 PyTorch 环境的步骤:

  1. 安装 Anaconda:Anaconda 是一个开源的 Python 发行版本,包含了大量的科学计算库和工具。可以从 Anaconda
    官网下载对应操作系统的安装包并安装。
  2. 创建虚拟环境:为了避免不同项目之间的依赖冲突,可以创建一个虚拟环境。可以使用 Anaconda 中的 conda 命令创建一个名为
    myenv 的虚拟环境,命令如下:
conda create -n myenv python=3.9 

其中,myenv 为环境名称,python=3.9 表示使用 Python 3.9 版本。创建完毕后,可以使用以下命令激活虚拟环境:

conda activate myenv 
  1. 安装 PyTorch:可以通过 conda 或 pip 安装 PyTorch。如果需要使用 GPU 加速,需要安装对应的 CUDA 和 cuDNN。
    • 使用 conda 安装:
conda install pytorch torchvision torchaudio -c pytorch 

• 使用 pip 安装:

pip install torch torchvision torchaudio 
  1. 测试 PyTorch 安装:可以通过以下 Python 代码测试 PyTorch 是否安装成功:
import torch x = torch.rand(5, 3) print(x) 

运行以上代码,如果能够输出一个随机生成的张量,则表示 PyTorch 安装成功。

  1. 安装 Jupyter Notebook:Jupyter Notebook 是一个交互式的 Web 应用程序,可以在浏览器中编写和运行
    Python 代码。可以通过以下命令安装 Jupyter Notebook:
conda install jupyter 
  1. 启动 Jupyter Notebook:使用以下命令启动 Jupyter Notebook:
jupyter notebook 

在浏览器中打开 http://localhost:8888 可以进入 Jupyter Notebook 界面。
这些是搭建 PyTorch 环境的基本步骤。可以根据实际需求安装其他工具和库,如 NumPy、Pandas、Matplotlib 等。

  • 3
    点赞
  • 11
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

Hello.Reader

请我喝杯咖啡吧😊

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值