09 Ubuntu下安装pytorch

1.查看python版本

Ubuntu主界面下打开终端，执行命令

python --version

2.激活虚拟环境

终端执行命令

# 创建pytorch虚拟环境[pytorch为虚拟环境名称，指定python版本]
conda create -n pytorch python=3.8

# 激活环境
source activate pytorch

3.进入pytorch官网查看命令

pytorch官网：PyTorch

因为我的电脑没有独立显卡，所以我选择Linux - Conda - Python - CPU，复制run this command中命令行

 将命令行输入终端，输入y安装

4.验证

①安装完成后，在终端执行命令，进入python环境

python

②执行命令

import torch
print(torch.cuda.is_available())  -->  cpu版本显示false，gpu版本显示true

有显示说明成功运行安装pytorch

PS:删除创建的虚拟环境[pytorch为创建的虚拟环境名]

conda remove -n pytorch --all
# 上面方法不行的话，使用conda env remove -n pytorch