在服务器上配置pytorch

一.下载anaconda安装包

        在终端输入

        wget https://repo.anaconda.com/archive/Anaconda3-2020.07-Linux-x86_64.sh

        系统会自动将资源条下载到当前目录。

二.安装anaconda

        bash Anaconda3-2020.07-Linux-x86_64.sh -p PATH -u 

        安装过程输入yes或者enter

       注： PATH是你想要安装的位置。

        测试是否安装成功：

        source ~/.bashrc

        python

        到这里就安装成功了。 

        我们输入

        exit()

        退出python编辑环境。

        此时我们可以安装使用清华镜像源可以加快下载速度

(13条消息) Anaconda清华镜像源的使用及安装Pytorch失败问题解决_WannaSeaU的博客-CSDN博客_anaconda安装pytorch失败https://blog.csdn.net/WannaSeaU/article/details/88427010

三.创建并激活虚拟环境

        在终端输入这行命令

        conda create --name NAME python=3.7

        注：NAME为你想命名的虚拟环境

        现在我们已经创建好虚拟环境，进入我们自己的虚拟环境

        conda activate NAME

四.查看cuda的版本并下载对应的torch

        输入这条指令查看自己的CUDA版本

        cat /usr/local/cuda/version.txt

        找对应版本的torch    网址： Previous PyTorch Versions | PyTorch

        我的版本是10.0，打开网址找到对应的torch

        

         因为咱们用的是清华镜像源，从网站上复制过来的命令最后面去掉-c pytorch

        conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0

        安装完成之后测试一下

        输入python 进入编辑环境

        python

        在python编辑环境中输入

        import torch

        print(torch.__version__)

        print(torch.cuda.is_available())

        

        看见True就证明环境配好了