理工的小蔡同学

导航

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

如果是官网一步步来的就会是

 

 

 

 

 

 

 

      

posted on 2023-03-06 11:58  董舒尧呀  阅读(915)  评论(0编辑  收藏  举报