1.安装Anaconda(里面有pytorch+cuda+cudnn 一键安装)
去Anaconda官网下载对应版本
安装指令:
bash XXXX.sh
然后输入yes
然后一直enter阅读协议
安装位置可以自己修改
然后问是否添加环境变量路径:yes
输入 :anaconda-navigator
启动anaconda
安装完毕
清华的开源软件镜像站也提供了更换Anaconda仓库的镜像,在终端中输入以下命令
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 --set show_channel_urls yes
更改pytorch的源:
暂时没找到,清华源1.0版无效
2.安装pytorch
选择conda python环境(其实不影响,指令一样) CUDA 9.0
复制指令。
然后打开终端,输入: conda install pytorch torchvision cudatoolkit=9.0 -c pytorch
确定输入y 安装pytorch