Linux下如何安装,卸载pytorch,查看pytorch版本,更新conda版本(总结)
·
一、在Linux下安装pytorch
- 进入Linux环境以后,首先新建一个名为pytorch的虚拟环境,执行以下代码:
conda create -n pytorch python=3.6
- 激活新建的pytorch虚拟环境,执行以下代码:
source activate pytorch # conda版本比较低可以用这条代码
或者
conda activate pytorch # conda版本较新使用这条代码
- 使用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。

- 下载完成后,最后退出虚拟环境,执行以下代码:
source deactivate # conda版本比较低可以用这条代码
或者
conda deactivate # conda版本较新使用这条代码
二、在Linux下卸载pytorch
- 首先需要进入之前创建的名为pytorch的虚拟环境,执行如下代码:
source activate pytorch # conda版本比较低可以用这条代码
或者
conda activate pytorch # conda版本较新使用这条代码
- 使用conda卸载Pytorch,执行以下代码:(两个代码都要执行)
1.conda uninstall pytorch
2.conda uninstall libtorch
- 最后退出虚拟环境,执行以下代码:
source deactivate # conda版本比较低可以用这条代码
或者
conda deactivate # conda版本较新使用这条代码
三、在Linux下查看pytorch版本
- 首先进入python环境,在Linux终端里输入python进入python环境。
- 然后在python环境里,输入: import torch
- 然后在python环境里,输入:torch.version
- 最后退出python环境,输入:exit()
- 如下:
(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
魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。
更多推荐


所有评论(0)