当前位置:   article > 正文

ubuntu20.0.4下安装PyTorch_conda ubuntu安装pytorch

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

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/天景科技苑/article/detail/958836
推荐阅读
相关标签
  

闽ICP备14008679号