【深度学习环境之torch】为现有环境安装torch、torchvision、torchaudio,以cu117+python3.9为例
由于大多数帖子在编写安装教程时都是从安装显卡驱动和cuda开始。但是在现实应用中,大多数情况下电脑都已经安装完cuda和显卡驱动且代码需要在固定python版本下运行。
针对以上问题,本篇以cu117+python3.9为例介绍如何为现有环境安装pytorch等python库。本篇共介绍三种安装方式,分别是,pip安装、离线安装和conda安装。
总述(重要!!重要!!重要!!!)
由于torch与操作系统、python版本和cuda版本有关 ,所以无论是pip安装、离线安装还是conda安装,安装的流程均为
查看cuda版本->根据对应版本查找pytorch对应文件->下载安装包并安装
1、pip安装
作为常用的安装pytorch 的方法,使用pip安装主要包含以下几步:
1、查看电脑上的cuda版本(以linux系统为例)
使用如下指令查看自己电脑上的cuda版本:
nvcc-V
输入后出现以下界面,图中红框中的内容为电脑的cuda版本。
注意!!!!!!!在实际应用中,常使用如下指令查看电脑显卡的运行情况
nvidia-smi
图中标红处也显示cuda版本。但是,此cuda版本并不是系统安装的cuda版本,