力荐:win10安装cuda+cudnn+pytorch+TensorFlow超级简单的办法!

废话不多说,看步骤:

  1. 安装好anaconda,根据anaconda版本会默认一个python版本;
  2. 使用默认的python版本,或者新建一个指定版本的python新环境 conda create -n Pytorch python=3.6
  3. conda install tensorflow-gpu此命令会一键安装cudatoolkit、cudnn等最核心的环境。可以看看此命令运行后要安装的cudatoolkit、cudnn版本是否合乎要求,不合乎要求可以conda install tensorflow-gpu=1.x.x的方式指定cudatoolkit、cudnn的版本;
  4. 如果下载有问题,可以更换国内源,逐个执行以下命令;
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
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

由于清华镜像和中科大镜像在2019年4月相继停止使用,所以我们需要更换上交大的源,首先删除清华的源,r然后更换上交大的源: 清华源又可以用了,20200408更新

conda config --remove channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.sjtug.sjtu.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.sjtug.sjtu.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.sjtug.sjtu.edu.cn/anaconda/cloud/conda-forge/
  1. 安装pytorch,使用官网的指令pytorch官网 ,选择上面安装的版本,如果要历史版本的指令,点击这里;注意, -c pytorch表示从官方源下载pytorch,删除掉,才能使用交大的源下载。
    力荐:win10安装cuda+cudnn+pytorch+TensorFlow超级简单的办法!_第1张图片
    如果以上下载pytorch还是很慢,使用下面的方法,下载非常快,亲测有效
    首先,将安装包下载到本地;然后anaconda命令窗口中运行pip install d:/Dir/torch-1.1.0-cp36-cp36m-win_amd64.whl,然后就安装成功啦!
1)pytorch0.4.0, python3.5, Linux, cu80, 86_64 使用以下链接进行下载到本地
http://download.pytorch.org/whl/cu80/torch-0.4.0-cp35-cp35m-linux_x86_64.whl

2) pytorch1.1.0, python3.6, windows, cpu, 64 使用以下链接进行下载到本地
http://download.pytorch.org/whl/cpu/torch-1.1.0-cp36-cp36m-win_amd64.whl

3) pytorch1.1.0, python3.6, windows, gpu, 64 使用以下链接进行下载到本地
http://download.pytorch.org/whl/cu90/torch-1.1.0-cp36-cp36m-win_amd64.whl
==注:需要下载其他版本,手动更改链接即可==
  1. 安装完了pytorch,进行测试
import torch   # 导入torch包 如果没有报错则表示安装成功
torch.cuda.is_available()  # 返回True则表示GPU可用, 反之则不可用

你可能感兴趣的:(力荐:win10安装cuda+cudnn+pytorch+TensorFlow超级简单的办法!)