参考视频:远程服务器安装anaconda并配置pytorch环境
使用服务器运行项目,使用的是服务器的环境(即服务器的虚拟环境和其中的pytorch),与连接服务器的本机就是你的电脑应该是没有关系的。
链接:Anaconda官网
这是我下载的版本Anaconda3-2023.09-0-Linux-x86_64.sh
将下载好的文件传到xftp中
bash Anaconda3-2023.09-0-Linux-x86_64.sh
安装Anaconda后出现的conda: command not found问题
nvidia-smi
使用conda创建一个名为 pytorch
的虚拟环境,并指定Python版本为3.8。
conda create -n pytorch python=3.8
进入我们刚刚安装好的环境,我们要在pytorch
这个环境下安装,不要在base
环境下安装
conda activate pytorch
在pytorch官网查找符合自己cuda版本的pytorch
网页往下滑,点击previous versions of pytorch
我需要安装的是cuda版本为12.0 的pytorch,找了半天并没有CUDA12.0的版本,我选择的是CUDA 11.6
CUDA版本通常是向下兼容的。你可以在安装了CUDA 11.6的系统上使用适用于CUDA 12.0的PyTorch
conda install pytorch==1.13.1 torchvision==0.14.1 torchaudio==0.13.1 pytorch-cuda=11.6 -c pytorch -c nvidia
在终端输入,等待运行
python
import torch
torch.cuda.is_available()
出现True表示安装成功