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

相关推荐

  1. ubuntu20.0.4安装PyTorch

    2024-06-15 22:04:01       31 阅读
  2. Ubuntu Docker安装 2024

    2024-06-15 22:04:01       28 阅读
  3. Ubuntu安装 Dotnet

    2024-06-15 22:04:01       41 阅读
  4. UbuntuClickHouse安装

    2024-06-15 22:04:01       31 阅读

最近更新

  1. docker php8.1+nginx base 镜像 dockerfile 配置

    2024-06-15 22:04:01       98 阅读
  2. Could not load dynamic library ‘cudart64_100.dll‘

    2024-06-15 22:04:01       106 阅读
  3. 在Django里面运行非项目文件

    2024-06-15 22:04:01       87 阅读
  4. Python语言-面向对象

    2024-06-15 22:04:01       96 阅读

热门阅读

  1. 钉钉Stream模式推送程序环境部署

    2024-06-15 22:04:01       28 阅读
  2. 图神经网络工具篇

    2024-06-15 22:04:01       25 阅读
  3. C#多线程与函数对象的实例

    2024-06-15 22:04:01       22 阅读
  4. Elasticsearch与Kafka集成:实现数据流处理

    2024-06-15 22:04:01       30 阅读
  5. 代码随想录Day59

    2024-06-15 22:04:01       29 阅读
  6. lvgl手势事件判断为点击事件问题

    2024-06-15 22:04:01       28 阅读
  7. 解决测试类Class not found报错

    2024-06-15 22:04:01       31 阅读
  8. 区块链之快照

    2024-06-15 22:04:01       29 阅读
  9. web开发的尽头是servlet

    2024-06-15 22:04:01       32 阅读
  10. 反射,枚举以及lambda表达式

    2024-06-15 22:04:01       22 阅读
  11. webpack和vite区别

    2024-06-15 22:04:01       24 阅读
  12. Vue待学习

    2024-06-15 22:04:01       22 阅读