一.安装anaconda
进入mobaxterm界面后选择使用的节点然后登陆,登录后输入命令python -V(V大写)来查看当前的python版本,根据当前的版本去清华源下载对应的anaconda
清华源地址:Index of /anaconda/archive/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirrorp
python3.8可用的anaconda版本有:(注意要linux x86_64版本)
下载完成后直接拖拽到下面位置:
输入命令bash Anaconda3-2020.02-Linux-x86_64.sh 进行安装,一路enter或输入yes即可
二.创建虚拟环境配置pytorch
输入命令conda create -n pytorch python=3.8创建环境
成功安装后先输入命令source activate激活,然后输入conda activate pytorch激活刚才新建的pytorch环境
输入命令nvidia-smi查看cuda版本
去pytorch官网Start Locally | PyTorch按照自己的版本生成对应的语句然后输入该语句命令安装即可
检查是否安装成功:新建一个.py文件写以下两句:
import torch
if torch.cuda.is_available():
print('ok')
退出当前虚拟环境输入命令:conda deactivate即可