一. 环境搭建
1. cuda
版本地址:https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html下载地址:https://developer.nvidia.com/cuda-toolkit-archive环境变量:~\NVIDIA corporation\NVSMIcmd验证:nvidia-smi
2. anaconda
anaconda的清华镜像网:https://mirror.tuna.tsinghua.edu.cn/help/anaconda/环境变量:~\~\Scriptscmd验证:conda env list修改conda源:conda config --set show_channel_urls yesvim C:\Users\MECHREVO_Z2_AIR\.condarc替换成如下内容:channels: - defaultsshow_channel_urls: truedefault_channels: - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2custom_channels: conda-forge: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud msys2: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud bioconda: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud menpo: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud pytorch: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud pytorch-lts: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud simpleitk: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud创建虚拟环境(需要关闭翻墙代理):conda create -n zhn-vir删除虚拟环境:conda remove –n zhn-vir –-all激活环境:activate zhn-vir返回默认环境:deactivate
二. 安装
1. 下载中断
迅雷下载后conda clean --allconda install --use-local .\***.tar.bz2
2 pytorch
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
3. pip
pip配置文件位置:C:\Users\zhanghenan\AppData\Roaming\pip[global]index-url = http://mirrors.aliyun.com/pypi/simple/trusted-host = mirrors.aliyun.com
三 启动
python demo_toolbox.py -d .\samples