当前位置:   article > 正文

windows10下pytorch1.4.0安装_pytorch 1.4.0

pytorch 1.4.0

安装步骤如下:
1.打开Anaconda Prompt,输入:

conda create -n pytorch1.4.0 python==3.6

创建一个pytorch的虚拟环境。

2.激活环境:

conda activate pytorch1.4.0

3.安装CUDA和Cudnn

conda install cudatoolkit==10.1

conda install cudnn==7.6.5

如果网不好的可以下载离线包:
cudatoolkit10.1:
链接:https://pan.baidu.com/s/1V9QVXn0E9uruWKwk9CDRuQ
提取码:1234

cudnn7.6.5:
链接:https://pan.baidu.com/s/1rGshlMTZgKNzbccMvXET6g
提取码:1234

下载之后放在创建的虚拟环境之下:
在这里插入图片描述
然后安装:

conda install cudatoolkit-10.1.168-0.tar.bz2
conda install cudnn-7.6.5-cuda10.1_0.tar.bz2

在这里插入图片描述
在这里插入图片描述

4.下载pytorch本地包
在这里插入图片描述
链接:https://pan.baidu.com/s/1ayKaBMlzFijudYtTxWD28A
提取码:1234

在安装之前,先安装依赖包:

pip install numpy -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install six -i https://pypi.tuna.tsinghua.edu.cn/simple

把刚才下载的包放在虚拟环境之下:
在这里插入图片描述
然后安装:

pip install torch-1.4.0-cp36-cp36m-win_amd64.whl
pip install torchvision-0.5.0-cp36-cp36m-win_amd64.whl --no-deps

注意安装torchvision的时候要加上 –no-deps
然后等待安装完毕即可。

5.测试是否安装成功
输入python进入Python的环境,
然后输入import torch,如果没报错即安装成功。
如果是python3.6.0可能会遇到这个错误:
在这里插入图片描述
这时候可以输入quit()退出Python环境,然后升级一下python版本:

conda install python==3.6.5
  • 1

等待安装完毕。

然后再进入python环境,同样import torch,没报错则继续:

print(torch.cuda.is_available())
  • 1

最后返回true即说明GPU可用,安装成功。

本文内容由网友自发贡献,转载请注明出处:【wpsshop博客】
推荐阅读
相关标签
  

闽ICP备14008679号