Linux系统下通过Anaconda安装Pytorch

神经网络   2025-01-01 09:27   29   0  

一、创建虚拟环境

切换到conda用户,执行:

conda create -n pytorch python=3.10(pytorch可自行命名)

如后续需要安装 TensorFlow 2.x 版本,Python 推荐使用 3.7 到 3.10 版本。

创建过程:

54804_brqi_1836.png

Proceed输入y,点击回车Enter:

54938_bwme_5769.png

出现done提示:

55017_6lkt_9806.png

二、激活环境

conda activate pytorch

出现以下提示:

55077_acrp_7089.png

检测环境是否安装好:

conda info --envs

55125_le0h_6658.png

选择适合自己的Pytorch版本,点击https://pytorch.org/

根据自己的实际情况选择:

55186_egou_6998.png

复制这句话,输入到控制台执行:

conda install pytorch torchvision torchaudio cpuonly -c pytorch

55354_pryi_1763.png

55360_e8dn_7158.png

55367_gorq_6528.png

55376_kcot_2272.png

弹出提示,输入 y:

55436_qlga_2751.png

点击回车Enter:

55505_hkzi_8344.png

显示done,完成安装:

55567_gqbp_1793.png

三、测试安装成功

输入:python

55698_mwa5_2522.png

输入:import torch

55703_mctx_2174.png

执行 exit() 退出之后,执行以下命令查看自己安装的环境:

conda info --envs

55709_idtw_8590.png


博客评论
还没有人评论,赶紧抢个沙发~
发表评论
说明:请文明发言,共建和谐网络,您的个人信息不会被公开显示。