【FedML】环境配置

FedML环境配置

step1：打开cmd

step2：创建一个环境

conda create -n fedml python==3.8

step3：启动环境

activate fedml

step4：安装FedML

pip install fedml

step5：安装指定版本 Pytorch

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

ps：需要调用gpu可把命令换成这句话

对应cuda版本11.6

pip install torch==1.13.1+cu116 torchvision==0.14.1+cu116 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu116 -i https://mirrors.aliyun.com/pypi/simple/

对应cuda版本10.2

conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.2 -c pytorch

step6：

pip install imageio

最后对环境进行验证

python

import torch

torch.cuda.is_available()

需要结果返回true

若返回false则卸载安装包

pip uninstall torch

pip uninstall torchvision

pip uninstall torchaudio

再 去pytorch网站找到cuda对应版本的命令

Previous PyTorch Versions | PyTorch

若需要linux新账号需要下载anaconda可以参考下面的博客：

如何在Linux服务器上安装Anaconda（超详细）_linux安装anconda_流年若逝的博客-CSDN博客

查看环境命令;conda info -e

卸载环境命令：conda remove -n fedml --all  （注：fedml为环境名字）

查看是否有anaconda：conda --version

查看gpu:nvidia-smi

注：RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED 错误就是cuda版本的pytorch的一系列包版本不匹配导致