赞
踩
具体版本配置还需要据根自身情况
我的是python3.7,cuda10.1 cudnn-7.6.5
https://www.pythonf.cn/read/135888
之前都是按照官网命令来安装,但是有时候官网这个下载速度实在是慢得让人受不了,这里给出另一种安装方式,用安装包来安装pytorch,利用清华镜像源和anaconda。
首先创建一个虚拟环境并激活
- conda create -n torch python=3.6
- conda activate torch
然后在清华源头下载安装包
https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
根据自己的需要,选择Linux系统还是win-64,进入之后可选取pytorch的版本以及cpu版还是gpu版
例如:我选择linux-64,然后选择python3.6、pytorch1.1.0、cuda10
同时还需要torchvision的话,可以选择python3.6、cuda10、torchvision0.3.0
安装包下好之后,选择好路径执行:
- conda install --offline pytorch-1.1.0-py3.6_cuda10.0.130_cudnn7.5.1_0.tar.bz2
-
- conda install --offline torchvision-0.3.0-py36_cu10.0.130_1.tar.bz2
这样就在torch的虚拟环境下装好了pytorch和torchvision,最后可能还需要自己安装cuda和cudnn,注意版本匹配
- conda install cudatoolkit=10.0
-
- conda install cudnn
这种方法就不用再担心网速的限制了,走清华源来下载速度非常快!
我的具体步骤:
测试是否安装成功:
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。