win10安装pytorch

发布于:2022-12-22 ⋅ 阅读:(258) ⋅ 点赞:(0)

1、安装pytorch需要提前安装好anaconda

配置anaconda镜像源:

      conda config --add channel http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/win-64/

       conda config --add channel  http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/win-64/

       conda config --add cannel http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64

       conda config --set show_channel_urls yes

查看当前使用源:

conda config --show-sources

运行 conda clean -i 清除索引缓存,保证用的是镜像站提供的索引。

还原原始源:conda config --remove-key channels

2、打开 Anaconda 自带的 Anaconda Prompt:创建虚拟环境

先查看安装的python版本:

conda create --name pytorch python=3.9

注意,这里的 pytorch 是虚拟环境的名称,可随意取。

创建好后会提示:

3、进入pytorch虚拟环境:

activate pytorch

退出虚拟环境:

conda deactivate

4、安装pytorch

4.1 先查看当前计算机的cuda版本(英伟达版本):nvidia-smi

4.2 选择对应的pytorch版本安装

Start Locally | PyTorch

注意:由于cuda版本是11.4在这里就只能选cuda 11.3的版本进行安装

安装命令:

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

注意:如果安装过程中报HTTP 000 之类的错误,可以把上面安装命令中的-c去掉。

4.3 安装过程中报错

解决方法:问题出在源的问题上,源文件在C:\Users\zyl\.condarc

把里面的https改成http,并把“- defaults”删除。或者可以直接用下面的源:(复制到.condarc文件里面)

channels:
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/win-64
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/win-64
show_channel_urls: true

注意1:如果安装错误了,需要把整个虚拟环境都删除,然后再重新安装:

进入anaconda的控制管理器,选择环境选项--当前虚拟环境pytorch,点击左下角的remove

注意2:在安装pytorch的时候最好不要再点安装界面,这样很容易下载安装包的时候断掉(亲测多次)

5、检验是否安装成功pytorch

能够检验是否成功安装pytorch, cuda, cudnn,并能输出GPU型号

import torch

torch.cuda.is_available()

x=torch.randn(4,4)

print(x)

返回结果如上图,则说明整个过程安装完成。

本文含有隐藏内容,请 开通VIP 后查看