用官网的命令太慢,换国内的源也不行,因为国内的源会默认给你安装cpu版本,所以,,,还是往下看吧
一、如果还没安装Anaconda和Cuda的先去这个链接(Windows 10安装Tensorflow-gpu及cuda+cuddn详细图解(刚装完赶紧做笔记:作经验分享)_我系渣渣的博客-CSDN博客)安装前三步后再来安装Pytorch。
二:创建一个独有的虚拟环境
1、创建 conda 环境
启动 Anaconda prompt
conda create -n xxxx pip python=3.6 #xxxx是你虚拟环境的名,想要啥就啥,python版本号可换
2、激活创建的 conda 环境,进入创建的虚拟环境
conda activate xxxx
二、开始安装pytorch-gpu
- 这里建议先离线下载好再安装,因为用官网的命令安装总是容易安装成cpu版本的,我们到这个网站 https://download.pytorch.org/whl/torch_stable.html 下载好所需要的包,然后pip安装
- 也可以到打开清华源下的pytorch库
https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/
- 找到自己合适的版本,比如我需要GPU版本的并且支持cuda10,网页ctrl+f快捷搜索,找到文件中
- pytorch-1.2.0-py3.7_cuda100_cudnn7_1.tar.bz2
版本号:1.2.0 ------根据自己需求更改
-
nightly版本可以看作是开发版,一般不选用,
命名格式为pytorch-版本-python版本-cuda表示gpu版-cudnn版本.tar.bz2
大小一般为5、600M -
下载后,放到合适的盘,接着在刚才创建的那个虚拟环境下,cd到文件目录
输入命令 -
conda install --offline pytorch-1.2.0-py3.7_cuda100_cudnn7_1.tar.bz2 #文件全称
安装完pytorch后,直接输入命令安装完依赖的包:
-
conda install torchvision
打开conda中的python进行测试
-
import torch print(torch.cuda.is_available()) print(torch.version.cuda)
输入结果为True和10.0就对了