pytorch环境配置

Anaconda的安装

网址:Anaconda | The World’s Most Popular Data Science Platform->Free Download->python 3.11下载

检查Anaconda是否安装成功

在开始里找到Anaconda Prompt,前面有(base)说明安装成功

环境管理

Anaconda使用虚拟环境安装两个版本的pytorch0.4、1.0的步骤:

conda create -n pytorch python=3.11
​
conda activate pytorch(切换)
​
pip list(查看里面的包)
pytorch的安装
查看自己显卡型号

方式1.任务管理器——性能

方式2.右击“此电脑”——属性——设备管理器——显示适配器

查看自己的显卡版本是否支持GPU:CUDA GPUs - Compute Capability | NVIDIA Developer
下载pytorch

网址:PyTorch

最后把run this command的命令复制到Anaconda Prompt中下载就可以了

下载失败

解决conda安装pytorch:An HTTP error occurred when trying to retrieve this URL.

原因:默认使用外国的下载源,所以下载失败,使用清华源下载。

添加清华源镜像,在Anaconda中输入

conda config --set show_channel_urls yes

打开:C:\Users\29337\.condarc替换

channels:
  - defaults
show_channel_urls: true
default_channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
custom_channels:
  conda-forge: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  msys2: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  bioconda: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  menpo: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  pytorch: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  simpleitk: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud

保存后关闭该文件,打开Anconda Prompt 输入:

conda clean -i #清除

在Anaconda Prompt中输入刚刚在run this command里复制的命令:

conda install pytorch torchvision torchaudio cpuonly -c pytorch

检查是否安装好pytorch

import torch
torch.cpu.is_available()/torch.gpu.is_available()

最后安装成功

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值