centos 显卡驱动、cuda、cudnn安装参考地址:
1.centos7 安装显卡驱动、cuda、cudnn-CSDN博客
1.安装anaconda
步骤1:下载
下载地址:
Index of /
下载版本
步骤2:运行
sh Anaconda3-2021.05-Linux-x86_64.sh
一路执行回车,然后选择yes,最后安装完成
步骤3:测试
如果没有出现anacoda,则添加环境变量
vim ~/.bashrc
添加anaconda的地址
使环境变量生效,source ~/.bashrc
再执行python命令测试
新建空间:
conda create -n swin python=3.8
conda activate swin
步骤4:安装包
pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html
由于在线安装总出错,建议先将包下载下来之后,再安装。由于torch,依赖numpy,所以要先安装numpy。
pip install numpy -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com
pip install torch-1.8.0+cu111-cp38-cp38-linux_x86_64.whl --trusted-host pypi.douban.com
安装torchvision依赖于pillow
pip install pillow -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com
pip install torchvision-0.9.0+cu111-cp38-cp38-linux_x86_64.whl --trusted-host pypi.douban.com
pip install torchaudio-0.8.0-cp38-cp38-linux_x86_64.whl --trusted-host pypi.douban.com
步骤5:测试
返回True表示成功。
步骤6:安装其他的包
pip install cython matplotlib opencv-python