Cuda环境配置
export PATH="/usr/local/cuda-11.0/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda-11.0/lib64:$LD_LIBRARY_PATH"
pytorch
官网
这里贴一下cuda 11.7对应的torch(gpu)
conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia
# 官网pip命令似乎是挂了
matplotlib库
pip install matplotlib -i https://pypi.tuna.tsinghua.edu.cn/simple
该过程不仅安装了 matplotlib ,还安装了依赖的 numpy、python-dateutil、kiwisolver 、 pillow 、pyparsing 、cycler 、six 库。
清华源安装cv2、tqdm、face_alignment、dlib
* pip install -i https://pypi.tuna.tsinghua.edu.cn/simple opencv-python;
* pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tqdm;
* pip install -i https://pypi.tuna.tsinghua.edu.cn/simple face_alignment;
* pip install -i https://pypi.tuna.tsinghua.edu.cn/simple dlib
换源
-i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com
cupy
pip install cupy-cuda117 -i https://pypi.tuna.tsinghua.edu.cn/simple/
#(后面的117对应cuda版本
yaml
pip install pyyaml
mmcv
pip install -U openmim
mim install mmcv
torchsummary
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple torchsummary
此处库都可以直接安装
conda install numba
pip install pytorch-msssim
pip install skimage.metric
conda 克隆环境
conda create -n B(新环境的名称) --clone A(想克隆的环境名称)
conda删除环境
conda remove -n torch --all