Windows Torch 安装
首先,电脑要有显卡(没有显卡建议查cpu版本Torch安装和使用)
一、基础装备
(一)、Pycharm 下载地址:Download PyCharm: Python IDE for Professional Developers by JetBrains
(二)、Anaconda 下载地址:Anaconda | The World's Most Popular Data Science Platform
(三)、CUDA和Cudnn(我觉得下载10.2版本的CUDA比较方便)
CUDA下载地址:CUDA Toolkit Archive | NVIDIA Developer
Cudnn下载地址:cuDNN Archive | NVIDIA Developer
基础装备的安装先留个坑,有空再记录(应该很简单),目前网上一找一大堆哈哈
二、正文PyTorch安装
(一)、我尝试了直接命令行pip install torch,用的是清华源镜像,但是安装下来的都是cpu版本,所以就不用直接安装法了
(二)、官网安装(官网:PyTorch)
版本选择
历史版本选择选择后下拉至Wheel然后选择对应版本(我这是10.2的CUDA)
然后复制网址打开!然后复制网址打开!然后复制网址打开!
打开网址后选择要下载的Torch和Torchvision
在torch网页按Ctrl+F搜索cu102,然后找到安装的Anaconda对应Python的版本(我的是3.9,查看用cmd输入↓)
conda activate
python --version
注意:Linux版本很多,要找到win版本的!
然后同理下载对应版本torchvision(版本对应看:PyTorch中torch、torchvision、torchaudio版本对应关系_Liekkas Kono的博客-CSDN博客)
找到下载的两个whl文件,复制粘贴到\anaconda\Scripts(你找得到的地方)打开Anaconda Prompt并进到你放whl的文件夹
pip install "torch-1.10.2+cu102-cp39-cp39-win_amd64.whl" pip install "torchvision-0.11.3+cu102-cp39-cp39-win_amd64.whl"
大功告成,安装完成
检测一下
python
import torch
torch.cuda.is_available()
只要返回结果是True就是安装成功了
假设你是直接pip安装的(清华源还是官网)
你输入
python
import torch
torch.__version__
返回应该是cpu
如果是官网一步步来的就会是