Linux杂谈 | (8) Anaconda下安装Pytorch-GPU版

目录

0. 确保有网

1. 安装Anaconda 

2. 新建Pytorch虚拟环境

3. 激活Pytorch虚拟环境

4. 查看cuda版本

5. 在Pytorch环境下安装torch

6. 测试

---

0. 确保有网

首先确保服务器有网络连接,执行命令：

ping www.baidu.com

出现下图，意味着有网：

1. 安装Anaconda 

具体安装Anaconda的操作，可以查看我的另一篇博客：在Linux服务器上安装Anaconda

 

2. 新建Pytorch虚拟环境

conda create -n Pytorch python=3.6
# 环境名称：Pytorch python版本：3.6

3. 激活Pytorch虚拟环境

conda activate Pytorch 或
source activate Pytorch

更多有关Anaconda环境的操作命令，可以查看我的另一篇博客：https://blog.csdn.net/sdu_hao/article/details/104324397

 

4. 查看cuda版本

cat /usr/local/cuda/version.txt 

更多Linux服务器相关信息的查看命令见我的另一篇博客：查看Linux服务器各种相关信息

 

5. 在Pytorch环境下安装torch

打开Pytorch起始页，根据所使用Linux服务器的信息，一一进行选择：

选择完后，在命令行执行上图红框中的命令即可。

如果当前的选择项，没有适合所使用的Linux服务器的配置，比如之前查到所使用的Linux服务器的cuda版本为9.0，此时可以点击上图中篮框中的链接，安装之前版本的Pytorch。

执行上图篮框中的命令即可，可能需要一些时间。

 

6. 测试

出现上图中的情况，则安装成功。