ubuntu20.0.4下安装PyTorch

参考文档
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。一般这个命令不报错的话就证明安装成功。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值