创建虚拟环境:conda create -n your_env_name python=3.6
激活虚拟环境:source activate your_env_name
添加conda国内镜像:
-
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
安装cuda:conda install cudatoolkit=9.0 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/linux-64/
安装cudnn:conda install cudnn=7.4.1 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/linux-64/
安装pytorch指定版本:conda install pytorch=0.3.0 torchvision=0.2.0 -c soumith
(注意pytorch与cuda版本匹配)
切换cuda 版本命令:
sudo rm -rf /usr/local/cuda
sudo ln -s /usr/local/cuda-9.0 /usr/local/cuda
source ~/.bashrc
nvcc -V #查看cuda版本
(有道云笔记的命令 直接复制终端居然不能识别 坑