win10使用pip安装pytorch-gpu

首先,我的电脑之前一直用的tensorflow-gpu,所以python、cuda、cudnn这些环境都已经安装好了。笔记本比较旧了,安装的cuda是8.0版本。

下面开始使用pip安装pytorch-gpu:

1、打开官网 https://pytorch.org/get-started/locally/ 

      如果你的CUDA版本比较高可以直接按照下图选定安装

win10使用pip安装pytorch-gpu_第1张图片

如果像我一样CUDA8.0,  那就选择之前的版本

win10使用pip安装pytorch-gpu_第2张图片

  也提供了不同系统,conda和pip的不同安装命令,

win10使用pip安装pytorch-gpu_第3张图片

  我的python是3.5版本,所以选择这个。点击后,复制网址  https://download.pytorch.org/whl/cu80/torch-1.0.0-cp35-cp35m-win_amd64.whl

  2、命令提示符- pip install

        使用清华源换源,安装速度快

pip install https://download.pytorch.org/whl/cu80/torch-1.0.0-cp35-cp35m-win_amd64.whl -i https://pypi.tuna.tsinghua.edu.cn/simple some-package

win10使用pip安装pytorch-gpu_第4张图片

3、检验是否安装成功

win10使用pip安装pytorch-gpu_第5张图片

     大功告成!

4、安装torchvision

      torch1.0.0对应安装torchvision 0.2.0

pip install --no-deps torchvision==0.2.0

 

你可能感兴趣的:(安装环境,python,pip,pytorch)