当前位置：首页 > news >正文

ubuntu22.04下使用conda安装pytorch(cpu及gpu版本)

news 2025/9/16 4:58:06

本文介绍了conda下安装cpu、gpu版本的pytorch；并介绍了如何设置镜像源

ubuntu环境安装pytorch的CPU版本与GPU版本

系统：ubuntu22.04
显卡：RTX 3050
依赖工具：miniconda

确认环境

lsb_release -a
No LSB modules are available.
Distributor ID:	Ubuntu
Description:	Ubuntu 22.04.3 LTS
Release:	22.04
Codename:	jammy

$ nvidia-smi
Tue Feb 13 21:51:33 2024       
+---------------------------------------------------------------------------------------+
| NVIDIA-SMI 535.154.05             Driver Version: 535.154.05   CUDA Version: 12.2     |
|-----------------------------------------+----------------------+----------------------+
| GPU  Name                 Persistence-M | Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp   Perf          Pwr:Usage/Cap |         Memory-Usage | GPU-Util  Compute M. |
|                                         |                      |               MIG M. |
|=========================================+======================+======================|
|   0  NVIDIA GeForce RTX 3050 ...    Off | 00000000:02:00.0 Off |                  N/A |
| N/A   48C    P3               7W /  35W |    435MiB /  4096MiB |     11%      Default |
|                                         |                      |                  N/A |
+-----------------------------------------+----------------------+----------------------++---------------------------------------------------------------------------------------+
| Processes:                                                                            |
|  GPU   GI   CI        PID   Type   Process name                            GPU Memory |
|        ID   ID                                                             Usage      |
|=======================================================================================|
|    0   N/A  N/A      1854      G   /usr/lib/xorg/Xorg                          226MiB |
|    0   N/A  N/A      2225      G   /usr/bin/gnome-shell                         84MiB |
|    0   N/A  N/A      3199      G   ...irefox/2987/usr/lib/firefox/firefox       95MiB |
|    0   N/A  N/A     48808      G   ...resh,SpareRendererForSitePerProcess       21MiB |
+---------------------------------------------------------------------------------------+

$ conda --version
conda 23.11.0

使用conda安装pytorch（CPU版本）

注意：默认conda安装pytorch的是cpu版本，如需要安装GPU版本的，注意直接看后面

创建一个新conda环境

$ conda create -n myPyt

新创建的环境不包含任何依赖可以使用conda list查看一下

开始安装pytorch（当然，也可以前面在创建环境的同时，把依赖包一同时安装了）

$ conda install pytorch

为了方便验证同时安装ipython

IPython 是 Python 的原生交互式 shell 的增强版，可以完成许多不同寻常的任务，比如帮助实现并行化计算；主要使用它提供的交互性帮助，比如代码着色、改进了的命令行回调、制表符完成、宏功能以及改进了的交互式帮助

# 激活环境
$ conda activate myPyt
# 安装ipython
$ conda install ipython

验证一下pytorch环境

输入ipython，进入交互式环境，依次输入如下两条命令import torch、torch.cuda.is_available()

$ ipython
Python 3.11.7 (main, Dec 15 2023, 18:12:31) [GCC 11.2.0]
Type 'copyright', 'credits' or 'license' for more information
IPython 8.20.0 -- An enhanced Interactive Python. Type '?' for help.In [1]: import torchIn [2]: torch.cuda.is_available()
Out[2]: False

发现当前的版本是不cpu版本，不支持cuda加速的，我们查看一下依赖

$ conda list
# packages in environment at /home/bing/miniconda3/envs/myEnv:                                                                                                       
#                                                                                                                                                                    
# Name                    Version                   Build  Channel                                                                                                   
_libgcc_mutex             0.1                        main                                                                                                            
_openmp_mutex             5.1                       1_gnu                                                                                                            
blas                      1.0                         mkl                                                                                                            
bzip2                     1.0.8                h7b6447c_0                                                                                                            
ca-certificates           2023.12.12           h06a4308_0                                                                                                            
cffi                      1.16.0          py311h5eee18b_0                                                                                                            
filelock                  3.13.1          py311h06a4308_0                                                                                                            
fsspec                    2023.10.0       py311h06a4308_0                                                                                                            
gmp                       6.2.1                h295c915_3                                                                                                            
gmpy2                     2.1.2           py311hc9b5ff0_0                                                                                                            
intel-openmp              2023.1.0         hdb19cb5_46306                                                                                                            
jinja2                    3.1.3           py311h06a4308_0                                                                                                            
ld_impl_linux-64          2.38                 h1181459_1                                                                                                            
libffi                    3.4.4                h6a678d5_0                                                                                                            
libgcc-ng                 11.2.0               h1234567_1                                                                                                            
libgomp                   11.2.0               h1234567_1                                                                                                            
libprotobuf               3.20.3               he621ea3_0                                                                                                            
libstdcxx-ng              11.2.0               h1234567_1                                                                                                            
libuuid                   1.41.5               h5eee18b_0                                                                                                            
markupsafe                2.1.3           py311h5eee18b_0                                                                                                            
mkl                       2023.1.0         h213fc3f_46344                                                                                                            
mkl-service               2.4.0           py311h5eee18b_1  
mkl_fft                   1.3.8           py311h5eee18b_0  
mkl_random                1.2.4           py311hdb19cb5_0  
mpc                       1.1.0                h10f8cd9_1  
mpfr                      4.0.2                hb69a4c5_1  
mpmath                    1.3.0           py311h06a4308_0  
ncurses                   6.4                  h6a678d5_0  
networkx                  3.1             py311h06a4308_0  
ninja                     1.10.2               h06a4308_5  
ninja-base                1.10.2               hd09550d_5  
numpy                     1.26.3          py311h08b1b3b_0  
numpy-base                1.26.3          py311hf175353_0  
openssl                   3.0.13               h7f8727e_0  
pip                       23.3.1          py311h06a4308_0  
pycparser                 2.21               pyhd3eb1b0_0  
python                    3.11.7               h955ad1f_0  
pytorch                   2.1.0           cpu_py311h6d93b4c_0  
readline                  8.2                  h5eee18b_0  
setuptools                68.2.2          py311h06a4308_0  
sqlite                    3.41.2               h5eee18b_0  
sympy                     1.12            py311h06a4308_0  
tbb                       2021.8.0             hdb19cb5_0  
tk                        8.6.12               h1ccaba5_0  
typing-extensions         4.9.0           py311h06a4308_1  
typing_extensions         4.9.0           py311h06a4308_1  
tzdata                    2023d                h04d1e81_0  
wheel                     0.41.2          py311h06a4308_0  
xz                        5.4.5                h5eee18b_0  
zlib                      1.2.13               h5eee18b_0

发现确实没安装任何cuda库，而pytorch的版本我们也可以看到确实cpu版本pytorch 2.1.0 cpu_py311h6d93b4c_0

conda安装GPU版本的pytorch

如何安装gpu版本的pytorch呢？我们继续
我们查看一下pytorch可安装的版本

$ conda search pytorch
Loading channels: done
# Name                       Version           Build  Channel
...
pytorch                       1.13.1 gpu_cuda113py39h0809116_0  pkgs/main           
pytorch                       1.13.1 gpu_cuda113py39h09dffc6_0  pkgs/main           
pytorch                       1.13.1 gpu_cuda113py39h926b89d_1  pkgs/main           
pytorch                       1.13.1 gpu_cuda113py39hde3f150_1  pkgs/main           
pytorch                        2.0.1 cpu_py310hab5cca8_0  pkgs/main           
pytorch                        2.0.1 cpu_py310hdc00b08_0  pkgs/main           
pytorch                        2.0.1 cpu_py311h53e38e9_0  pkgs/main           
pytorch                        2.0.1 cpu_py311h6d93b4c_0  pkgs/main           
pytorch                        2.0.1 cpu_py38hab5cca8_0  pkgs/main           
pytorch                        2.0.1 cpu_py38hdc00b08_0  pkgs/main           
pytorch                        2.0.1 cpu_py39hab5cca8_0  pkgs/main           
pytorch                        2.0.1 cpu_py39hdc00b08_0  pkgs/main           
pytorch                        2.0.1 gpu_cuda118py310h7799f5a_0  pkgs/main           
pytorch                        2.0.1 gpu_cuda118py310he342708_0  pkgs/main           
pytorch                        2.0.1 gpu_cuda118py311h7668aad_0  pkgs/main           
pytorch                        2.0.1 gpu_cuda118py311hce0f3bd_0  pkgs/main           
pytorch                        2.0.1 gpu_cuda118py38h7799f5a_0  pkgs/main           
pytorch                        2.0.1 gpu_cuda118py38he342708_0  pkgs/main           
pytorch                        2.0.1 gpu_cuda118py39h7799f5a_0  pkgs/main           
pytorch                        2.0.1 gpu_cuda118py39he342708_0  pkgs/main           
pytorch                        2.1.0 cpu_py310hab5cca8_0  pkgs/main           
pytorch                        2.1.0 cpu_py310hdc00b08_0  pkgs/main           
pytorch                        2.1.0 cpu_py311h53e38e9_0  pkgs/main           
pytorch                        2.1.0 cpu_py311h6d93b4c_0  pkgs/main           
pytorch                        2.1.0 cpu_py38hab5cca8_0  pkgs/main           
pytorch                        2.1.0 cpu_py38hdc00b08_0  pkgs/main           
pytorch                        2.1.0 cpu_py39hab5cca8_0  pkgs/main           
pytorch                        2.1.0 cpu_py39hdc00b08_0  pkgs/main

我们可以看到，2.1.0版本的，Build列中，有点的cpu有的是gpu，那么如何安装时指定安装带gpu表示的版本的呢？

我们只需制定版本号的同时，指定build即可

$ conda create -n pyt-gpu
$ conda activate pyt-gpu
$ conda install pytorch=2.0.1=gpu_cuda118py39he342708_0
...
The following NEW packages will be INSTALLED:_libgcc_mutex      pkgs/main/linux-64::_libgcc_mutex-0.1-main _openmp_mutex      pkgs/main/linux-64::_openmp_mutex-5.1-1_gnu blas               pkgs/main/linux-64::blas-1.0-mkl ca-certificates    pkgs/main/linux-64::ca-certificates-2023.12.12-h06a4308_0 cffi               pkgs/main/linux-64::cffi-1.16.0-py39h5eee18b_0 cudatoolkit        pkgs/main/linux-64::cudatoolkit-11.8.0-h6a678d5_0 cudnn              pkgs/main/linux-64::cudnn-8.9.2.26-cuda11_0 cupti              pkgs/main/linux-64::cupti-11.8.0-he078b1a_0 filelock           pkgs/main/linux-64::filelock-3.13.1-py39h06a4308_0 gmp                pkgs/main/linux-64::gmp-6.2.1-h295c915_3 gmpy2              pkgs/main/linux-64::gmpy2-2.1.2-py39heeb90bb_0 intel-openmp       pkgs/main/linux-64::intel-openmp-2023.1.0-hdb19cb5_46306 jinja2             pkgs/main/linux-64::jinja2-3.1.3-py39h06a4308_0 ld_impl_linux-64   pkgs/main/linux-64::ld_impl_linux-64-2.38-h1181459_1 libffi             pkgs/main/linux-64::libffi-3.4.4-h6a678d5_0 libgcc-ng          pkgs/main/linux-64::libgcc-ng-11.2.0-h1234567_1 libgomp            pkgs/main/linux-64::libgomp-11.2.0-h1234567_1 libprotobuf        pkgs/main/linux-64::libprotobuf-3.20.3-he621ea3_0 libstdcxx-ng       pkgs/main/linux-64::libstdcxx-ng-11.2.0-h1234567_1 magma              pkgs/main/linux-64::magma-2.7.1-h2c23e93_0 markupsafe         pkgs/main/linux-64::markupsafe-2.1.3-py39h5eee18b_0 mkl                pkgs/main/linux-64::mkl-2023.1.0-h213fc3f_46344 mkl-service        pkgs/main/linux-64::mkl-service-2.4.0-py39h5eee18b_1 mkl_fft            pkgs/main/linux-64::mkl_fft-1.3.8-py39h5eee18b_0 mkl_random         pkgs/main/linux-64::mkl_random-1.2.4-py39hdb19cb5_0 mpc                pkgs/main/linux-64::mpc-1.1.0-h10f8cd9_1 mpfr               pkgs/main/linux-64::mpfr-4.0.2-hb69a4c5_1 mpmath             pkgs/main/linux-64::mpmath-1.3.0-py39h06a4308_0 ncurses            pkgs/main/linux-64::ncurses-6.4-h6a678d5_0 networkx           pkgs/main/linux-64::networkx-3.1-py39h06a4308_0 ninja              pkgs/main/linux-64::ninja-1.10.2-h06a4308_5 ninja-base         pkgs/main/linux-64::ninja-base-1.10.2-hd09550d_5 numpy              pkgs/main/linux-64::numpy-1.26.3-py39h5f9d8c6_0 numpy-base         pkgs/main/linux-64::numpy-base-1.26.3-py39hb5e798b_0 openssl            pkgs/main/linux-64::openssl-3.0.13-h7f8727e_0 pip                pkgs/main/linux-64::pip-23.3.1-py39h06a4308_0 pycparser          pkgs/main/noarch::pycparser-2.21-pyhd3eb1b0_0 python             pkgs/main/linux-64::python-3.9.18-h955ad1f_0 pytorch            pkgs/main/linux-64::pytorch-2.0.1-gpu_cuda118py39he342708_0 readline           pkgs/main/linux-64::readline-8.2-h5eee18b_0 setuptools         pkgs/main/linux-64::setuptools-68.2.2-py39h06a4308_0 sqlite             pkgs/main/linux-64::sqlite-3.41.2-h5eee18b_0 sympy              pkgs/main/linux-64::sympy-1.12-py39h06a4308_0 tbb                pkgs/main/linux-64::tbb-2021.8.0-hdb19cb5_0 tk                 pkgs/main/linux-64::tk-8.6.12-h1ccaba5_0 typing-extensions  pkgs/main/linux-64::typing-extensions-4.9.0-py39h06a4308_1 typing_extensions  pkgs/main/linux-64::typing_extensions-4.9.0-py39h06a4308_1 tzdata             pkgs/main/noarch::tzdata-2023d-h04d1e81_0 wheel              pkgs/main/linux-64::wheel-0.41.2-py39h06a4308_0 xz                 pkgs/main/linux-64::xz-5.4.5-h5eee18b_0 zlib               pkgs/main/linux-64::zlib-1.2.13-h5eee18b_0 
Proceed ([y]/n)?

安装时，从提示安装的依赖我们可以看出，这个版本确实带上了cuda相关包cudatoolkit、cudnn，这次安装显然时间长了很多，包的大小也近2G

安装完成后，我们再次确认一下cuda加速是否可用；

同样，我们先安装一个ipython

$ conda install ipython

进入ipython后，依次执行如下代码import torch、torch.cuda.is_available()

$ ipython
Python 3.9.18 (main, Sep 11 2023, 13:41:44) 
Type 'copyright', 'credits' or 'license' for more information
IPython 8.15.0 -- An enhanced Interactive Python. Type '?' for help.In [1]: import torchIn [2]: torch.cuda.is_available()
Out[2]: True

这次cuda可以正常工作，到此完成gpu版本的pytorch安装

当然不是采用conda安装的话，希望自己手鲁，从nvidia驱动、cuda、cudnn这些库开始手动一个个安装好后，最后再安装pytorch也是可以的；关于pytorch手动安装方式，这里给出官方地址，点这里

最后再安装个jupyter工具conda install jupyter
启动命令jupyter-notebook

ubuntu22.04下使用conda安装pytorch(cpu及gpu版本)

本文介绍了conda下安装cpu、gpu版本的pytorch；并介绍了如何设置镜像源 ubuntu环境安装pytorch的CPU版本与GPU版本系统：ubuntu22.04 显卡：RTX 3050 依赖工具：miniconda 确认环境 lsb_release -a No LSB modules are available.…...

编程日记 2024/2/19 0:45:38

突破编程_C++_高级教程（模板编程的基础知识）

1 模板编程的基本概念 C 的模板编程是一种编程技术，它允许程序员编写处理不同类型数据的通用代码。通过使用模板，可以创建与特定数据类型无关的函数或类，这些函数或类在编译时可以根据需要生成特定数据类型的版本。这增加了代码的复用性、灵…...

编程日记 2024/2/19 0:43:37

胆小勿入！AI创作恐怖电影宣传片《生化危机：重生》

胆小勿入！AI创作恐怖电影宣传片《生化危机：重生》 "The city is falling, and the dead walk among us." "In the shadow of the apocalypse, the fight for survival begins." "The streets are silent, but the nightmare …...

编程日记 2024/2/19 0:42:36

HTTP 超文本传送协议

1 超文本传送协议 HTTP HTTP 是面向事务的 (transaction-oriented) 应用层协议。使用 TCP 连接进行可靠的传送。定义了浏览器与万维网服务器通信的格式和规则。是万维网上能够可靠地交换文件（包括文本、声音、图像等各种多媒体文件）的重要基础。 H…...

编程日记 2024/2/19 0:41:35

MySQL导入/导出数据

MySQL导入/导出数据文章目录 MySQL导入/导出数据一、MySQL 导入数据1、mysql 命令导入2、source 命令导入3、使用 LOAD DATA 导入数据4、使用 mysqlimport 导入数据4.1、mysqlimport的常用选项介绍二、MySQL 导出数据1、使用 SELECT ... INTO OUTFILE 语句导出数据2、mysqldu…...

编程日记 2024/2/19 0:39:33

Matplotlib初探：认识数据可视化与Matplotlib

Matplotlib初探：认识数据可视化与Matplotlib Fig.1 利用Matplotlib进行数据可视化( 可视化代码见文末) 🌵文章目录🌵 🌳引言🌳🌳一、数据可视化简介🌳🌳二、Matplotlib库简介&#x…...

编程日记 2024/2/19 0:37:30

LeetCode 0987.二叉树的垂序遍历：遍历时存节点信息，遍历完自定义排序

【LetMeFly】987.二叉树的垂序遍历：遍历时存节点信息，遍历完自定义排序力扣题目链接：https://leetcode.cn/problems/vertical-order-traversal-of-a-binary-tree/ 给你二叉树的根结点 root ，请你设计算法计算二叉树的垂序遍历…...

编程日记 2024/2/19 0:36:29

TCP 和 UDP的区别

文章目录概述区别UDPTCPTCP与UDP的选择UDP和TCP编程区别概述 TCP（Transmission Control Protocol，传输控制协议）和 UDP（User Datagram Protocol，用户数据报协议）是互联网中两种最常用的传输层协议总的来…...

编程日记 2024/2/19 0:34:28

Python 将一维数组或矩阵变为三维正文正文话不多说直接上代码： import numpy as npsampling_points 10001arr np.linspace(0, 2, sampling_points) arr_3D arr.reshape(1, 1, -1) print(arr_3D) """ result: [[[0.0000e00 2.0000e-04 4.0000…...

编程日记 2024/2/19 0:31:25

Python如何实现定时发送qq消息

因为生活中老是忘记各种事情，刚好又在学python，便突发奇想通过python实现提醒任务的功能（尽管TIM有定时功能），也可定时给好友、群、讨论组发送qq消息。其工作流程是：访问数据库提取最近计划——>根据数据…...

编程日记 2024/2/19 0:28:21

支付方式接入：支付宝、微信支付、微软支付

支付方式接入：支付宝、微信支付、微软支付 1、微信支付-接入指引 2、支付宝-接入指引 3、微软支付-接入指引 3.1、使用visual studio打包应用（发布到微软市场）：Package a desktop app from source code using Visual Studio -…...

编程日记 2024/2/19 0:26:19

C++中的互斥量

互斥量是一个类，互斥量的使用必须引入头文件#include <mutex>。互斥量就如同一把锁，在同一时间，多个线程都可以调用lock成员函数尝试给这把锁头加锁，但是只有一个线程可以成功给这把锁加锁，其他没有加锁成功的线…...

编程日记 2024/2/19 0:24:17

盲盒小程序开发

现如今，盲盒已经成为了市场上不可忽视的新型消费模式，并且也逐渐遍布在全球各地中。盲盒的种类商品也逐渐丰富完善，不在局限于性价比高的盲盒玩具、手办等，也发展到了美妆、电子、食品等行业，具有较大的实用性和收藏价…...

编程日记 2024/2/19 0:23:16

安装 Windows 10

1.镜像安装镜像安装:安装Windows 10 2.安装过程(直接以图的形式呈现) 选择专业版的等待安装即可...

编程日记 2024/2/19 0:21:15

C++文件操作-＞文本文件(-＞写文件、读文件)、二进制文件(-＞写文件、读文件)

#include<iostream> using namespace std; #include <fstream>//头文件包含 //文本文件写文件 void test01() { //1.包含头文件 fstream //2.创建流对象 ofstream ofs; //3.指定打开方式 ofs.open("test.txt", ios::out); //4.写…...

编程日记 2024/2/19 0:20:14

Mac相关问题

Mac 更新node版本第一步，先查看本机node.js版本： node -v 第二步，清除node.js的cache： sudo npm cache clean -f 第三步，安装 n 工具，这个工具是专门用来管理node.js版本的，别怀疑这个工具…...

编程日记 2024/2/19 0:15:08

Python爬虫之Splash详解

爬虫专栏：http://t.csdnimg.cn/WfCSx Splash 的使用 Splash 是一个 JavaScript 渲染服务，是一个带有 HTTP API 的轻量级浏览器，同时它对接了 Python 中的 Twisted 和 QT 库。利用它，我们同样可以实现动态渲染页面的抓取。 1. 功…...

编程日记 2024/2/19 0:13:07

Deep深度系统下载安装Beyond compare4

Beyond Compare 4下载和安装 1、在线安装 Debian, Ubuntu安装命令： wget https://www.scootersoftware.com/bcompare-4.4.6.27483_amd64.deb sudo apt update sudo apt install ./bcompare-4.4.6.27483_amd64.deb Redhat Enterprise Linux, Fedora, CentOS安装命令…...

编程日记 2024/2/19 0:12:06

Qt 使用QScintilla 编辑lua 脚本

需求： 利用QScintilla 编辑lua 脚本步骤： 1，下载 QScintilla Riverbank Computing | Download 2, 打开 src/qscintilla.pro 文件编译出 dll库 3，工程中引入这个库注意debug 模式必须加载debug 版本编译的库&#xff0…...

编程日记 2024/2/19 0:11:05

2022长安杯复现

案件情况某地警方接到受害人报案称其在某虚拟币交易网站遭遇诈骗，该网站号称使用“USTD 币”购买所谓的“HT 币”，受害人充值后不但“HT 币”无法提现、交易，而且手机还被恶意软件锁定勒索。警方根据受害人提供的虚拟币交易网站调取了对应…...

编程日记 2024/2/19 0:10:04

stm32G473的flash模式是单bank还是双bank？

今天突然有人stm32G473的flash模式是单bank还是双bank？由于时间太久，我真忘记了。搜搜发现，还真有人和我一样。见下面的链接：https://shequ.stmicroelectronics.cn/forum.php?modviewthread&tid644563 根据STM32G4系列参考手…...

编程新知 2025/9/14 12:04:54

3.3.1_1 检错编码（奇偶校验码）

从这节课开始，我们会探讨数据链路层的差错控制功能，差错控制功能的主要目标是要发现并且解决一个帧内部的位错误，我们需要使用特殊的编码技术去发现帧内部的位错误，当我们发现位错误之后，通常来说有两种解决方案。第一…...

编程新知 2025/9/9 4:26:53

visual studio 2022更改主题为深色

visual studio 2022更改主题为深色点击visual studio 上方的工具-> 选项在选项窗口中，选择环境 -> 常规 ，将其中的颜色主题改成深色点击确定，更改完成...

编程新知 2025/9/12 11:37:35

《基于Apache Flink的流处理》笔记

思维导图 1-3 章 4-7章 8-11 章参考资料源码： https://github.com/streaming-with-flink 博客 https://flink.apache.org/bloghttps://www.ververica.com/blog 聚会及会议 https://flink-forward.orghttps://www.meetup.com/topics/apache-flink https://n…...

编程新知 2025/9/14 7:52:42

MySQL中【正则表达式】用法

MySQL 中正则表达式通过 REGEXP 或 RLIKE 操作符实现（两者等价），用于在 WHERE 子句中进行复杂的字符串模式匹配。以下是核心用法和示例： 一、基础语法 SELECT column_name FROM table_name WHERE column_name REGEXP pattern; …...

编程新知 2025/9/13 11:37:54