一、在Linux下安装pytorch

  1. 进入Linux环境以后,首先新建一个名为pytorch的虚拟环境,执行以下代码:
conda create -n pytorch python=3.6
  1. 激活新建的pytorch虚拟环境,执行以下代码:
source activate pytorch
  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

二、在Linux下卸载pytorch

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

三、在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, 版本, python, 虚拟环境, conda, 卸载, pytorch

相关文章推荐

添加新评论,含*的栏目为必填