ubuntu20.0.4下安装PyTorch

作者 : admin 本文共1001个字,预计阅读时间需要3分钟 发布时间: 2024-06-15 共1人阅读

参考文档
https://datawhalechina.github.io/thorough-pytorch/%E7%AC%AC%E4%B8%80%E7%AB%A0/1.2%20PyTorch%E7%9A%84%E5%AE%89%E8%A3%85.html

1:安装Anaconda
登录Anaconda | Individual Edition,https://www.anaconda.com/download/success ,选择相应系统DownLoad,此处以ubuntu20.0.4为例,
下载的安装文件为Anaconda3-2024.02-1-Linux-x86_64.sh,安装之。

2,创建虚拟环境,本文的python版本为3.11.7,环境的名称为nbpytorch
conda create -n nbpytorch python==3.11.7

3,安装包,本文安装的包为pandas
conda install pandas

4,激活环境命令, 本文的环境的名称为nbpytorch
conda activate nbpytorch

5,其他命令
(i),卸载包
conda remove package_name
# 注:package_name 替换成对应的包的名称,eg: pandas
(ii),显示所有安装的包
conda list
(iii),删除虚拟环境命令
conda remove -n env_name –all 
# 注:env_name 替换成对应的环境的名称
(iv),退出当前环境
conda deactivate

6,安装PyTorch
(1):登录PyTorch官网 https://pytorch.org/
选择配置,本文的pytorch安装命令如下,在终端上执行:

conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

(2), 在终端上执行激活环境(如果已激活,本步骤省略)
conda activate nbpytorch

(3),检验是否安装成功
在上一步已激活环境的情况下,在终端执行命令python,然后输入下面的代码。
import torch
torch.cuda.is_available()

这条命令意思是检验是否可以调用cuda,如果我们安装的是CPU版本的话会返回False,
能够调用GPU的会返回True。一般这个命令不报错的话就证明安装成功。

本站无任何商业行为
个人在线分享 » ubuntu20.0.4下安装PyTorch
E-->