本文记录下来方便配置aistudio
百度提供的的配置:linux系统,cuda9.2,显存16G,内存32G
1.设置清华镜像:
注:前两个是给创建新环境用的,后三个给pytorch用的
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes
2.创建虚拟环境:
conda create -n pytorch python=3.6
切换环境:
source activate pytorch
查看环境信息:
conda info -e
3.安装pytorch
注意!在执行conda安装pytorch需要去除 -c pytorch
注:否则pytorch会默认channel安装,不用镜像
原来:
conda install pytorch torchvision cudatoolkit=9.2 -c pytorch
改成:
conda install pytorch torchvision cudatoolkit=9.2
4.安装opencv
conda install opencv
(或者:pip install opencv-python)
5.查看显卡信息
注:目的是查看显卡占用和显存信息
nvidia-smi
6.查看cuda版本
nvcc --version