Linux安装Pytorch

Anaconda环境搭建

1. 在Anaconda3官方下载对应的系统版本，本教程以64位linux_16.04系统为例。官网下载
   

2. 安装Anaconda

   进入自己的下载目录
   cd ~/Downloads
   使用以下命令进行安装：
   bash Annaconda...xxx.sh(文件名，文件名可以用tab健自动补全)
   

   选择yes
   进入到以下页面：
   
   到此就已经完成安装了，选择yes进行conda初始化
   
   安装成功
   

   新开一个终端输入python 测试是否Anaconda安装成功
   

   可见新打开的终端前有个（base），说明安装成功
   

输入conda -V可以查看当前conda版本

conda -V

创建pytorch虚拟环境

首先创建与自己服务器中python版本相匹配的pytorch虚拟环境,下载可能有点慢，更换源可能会快一点，更换源的博客很多。

conda create -n pytorch python=3.8

接着进入创建的环境中

conda activate pytorch #

然后去pytorch官网找到适合自己电脑配置的pytorch版本进行安装，这里我是安装GPU版本的，安装cpu版本的选择cpu就好

安装命令在官网上选好配置后会显示出来，复制进终端运行即可

conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia

在pytorch环境下输入逐次输入以下命令验证是否安装成功

python #检查python版本
import torch
torch.cuda.is_available()

到此，恭喜你已经成功安装了cuda 和 pytorch