用官网的命令太慢,换国内的源也不行,因为国内的源会默认给你安装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
3、查看虚拟环境列表
我们通过cmd或者Anaconda Prompt可以打开命令列表,两者都行
conda env list
conda info --envs
4、删除环境
conda remove -n xxxxx(名字) --all
二、开始安装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 #文件全称
第二个在线安装方法:
-
镜像地址: 清华镜像:https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/ # 输入安装命令 conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=10.2 -c https://mirrors.aliyun.com/anaconda/cloud/pytorch/win-64/ conda install pytorch==1.10.0 torchvision==0.11.0 torchaudio==0.10.0 cudatoolkit=10.2 -c https://mirrors.aliyun.com/anaconda/cloud/pytorch/win-64/ conda install pytorch==1.12.0 torchvision==0.13.0 torchaudio==0.12.0 cudatoolkit=10.2 -c https://mirrors.aliyun.com/anaconda/cloud/pytorch/win-64/ conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cudatoolkit=10.2 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/ # 经过测验,以上Pytorch的版本输入安装命令会自动替换为cpu版本的下载,1.10.1版本下载的是gpu的 conda install pytorch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1 cudatoolkit=10.2 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/
-
安装完pytorch后,直接输入命令安装完依赖的包:
-
conda install torchvision
打开conda中的python进行测试
-
import torch print(torch.cuda.is_available()) print(torch.version.cuda)
输入结果为True和10.0就对了