一、安装pytorch
1、安装Miniconda
curl -0 https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
sh Miniconda3-latest-Linux-x86_64.sh
2、使用conda自动安装pytorch和所需环境库
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c conda-forge
二、安装spyder-ide
cd ~/miniconda/bin
./pip3 install rtree
./pip3 install spyder
./pip3 install spyder-terminal
运行spyder:
cd ~/miniconda/bin
./spyder
三、检查pytorch安装
python3
>>> import torch
>>> x = torch.rand(5, 3)
>>> print(x)