如何在服务器上配置环境

1.连接服务器

服务器ip：202.119.84.91    port：60407 

2.安装Anaconda3,选择python3.7/3.8

wget https://repo.anaconda.com/archive/Anaconda3-2020.07-Linux-x86_64.sh
#清华镜像源 https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/
bash Anaconda3-2020.07-Linux-x86_64.sh

3.安装完成后检查是否安装成功

source ~/.bashrc（#重新启动环境变量，一定要输入这条命令！）
python

3.创建虚拟环境

#创建虚拟环境
conda create -n pythorch python=3.7
#激活虚拟环境
conda activate pytorch
#退出虚拟环境
conda deactivate
#删除虚拟环境
conda remove -n NAME --all

4.查看对应cuda版本

cat /usr/local/cuda/version.txt
# 另一种
nvidia-smi

 

根据pytorch官网 本地|派托奇 (pytorch.org)​​​​​​或者参考该博主(9条消息) Pytorch和cuda对应版本_SSPo的博客-CSDN博客_pytorch和cuda对应版本

 查看对应的torch版本并安装，支持向下兼容

我的是cuda 10.1 所以输入以下命令

# CUDA 10.1
conda install pytorch==1.7.0 torchvision==0.8.0 torchaudio==0.7.0 cudatoolkit=10.1 -c pytorch

5、测试

输入python 进入编辑环境 

import torch
 
print(torch.__version__)
 
print(torch.cuda.is_available())

分别输出安装对应的pytorch版本和True 即可