Linux下如何安装,卸载pytorch,查看pytorch版本,更新conda版本(总结)

本文详细介绍在Linux环境下如何安装、卸载PyTorch,并提供查看PyTorch版本及更新conda的方法。包括创建虚拟环境、根据CUDA版本选择合适的PyTorch版本等关键步骤。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

一、在Linux下安装pytorch

  1. 进入Linux环境以后,首先新建一个名为pytorch的虚拟环境,执行以下代码:
conda create -n pytorch python=3.6
  1. 激活新建的pytorch虚拟环境,执行以下代码:
source activate pytorch # conda版本比较低可以用这条代码
或者
conda activate pytorch # conda版本较新使用这条代码
  1. 使用conda安装pytorch的版本,执行以下代码:
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
这里需要说明一下:
   通过上面代码安装pytorch,安装的是最新版本。如果是在服务器上配置自己的环境,那么服
务器的cuda版本我们无法改动,所以我们需要根据服务器的cuda版本下载适配的pytorch版本。
关于如何查看版本对应问题,附上pytorch的官网地址,上面有对应cuda的pytorch版本:
[pytorch官网:](https://pytorch.org/get-started/previous-versions/)

比如说,我的cuda版本是10.2,那我就用红线所画的这条命令下载pytorch。
版本对应

  1. 下载完成后,最后退出虚拟环境,执行以下代码:
source deactivate # conda版本比较低可以用这条代码
或者
conda deactivate # conda版本较新使用这条代码

二、在Linux下卸载pytorch

  1. 首先需要进入之前创建的名为pytorch的虚拟环境,执行如下代码:
source activate pytorch # conda版本比较低可以用这条代码
或者
conda activate pytorch # conda版本较新使用这条代码
  1. 使用conda卸载Pytorch,执行以下代码:(两个代码都要执行)
1.conda uninstall pytorch
2.conda uninstall libtorch
  1. 最后退出虚拟环境,执行以下代码:
source deactivate # conda版本比较低可以用这条代码
或者
conda deactivate # conda版本较新使用这条代码

三、在Linux下查看pytorch版本

  1. 首先进入python环境,在Linux终端里输入python进入python环境。
  2. 然后在python环境里,输入: import torch
  3. 然后在python环境里,输入:torch.version
  4. 最后退出python环境,输入:exit()
  5. 如下:
(pytorch) [usr@admin1 anaconda3]$ python
Python 3.6.13 |Anaconda, Inc.| (default, Jun  4 2021, 14:25:59)
[GCC 7.5.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> torch.__version__  #(注意,这里是两条横线 _ _)
'1.9.0'
>>> exit()

四、在Linux下更新conda版本

conda update -n base -c defaults conda

五、查看所有虚拟环境列表

conda info --envs
# 或者
conda env list

六、给Python环境设置全局的镜像加速

# 阿里云镜像
pip config set global.index-url https://mirrors.aliyun.com/pypi/simple/
# 清华镜像
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值