Windows+pytorch深度学习框架安装

(liux系统ubuntu的可查看

1.安装anaconda(清华镜像

我选的这个anaconda3代表里面的是python3->然后一路next

在此处将第一位也打勾,可以自动为你添加路径,否则需要自己手动添加路径

2 .安装pytorch

选择好配置之后,复制命令

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

如果直接在源路径下下载,直接打开Anaconda powershell Prompt,先创建pytorch环境变量conda create -n pytorch,然后激活环境变量conda activate pytorch,然后输入复制的命令运行即可

但可能速度很慢,可以现在Anaconda powershell Prompt中配置镜像源,输入

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --add channels https://mirrors.sjtug.sjtu.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.sjtug.sjtu.edu.cn/anaconda/pkgs/free/

 创建pytorch环境变量conda create -n pytorch,然后激活环境变量conda activate pytorch, 删除安装指令中的-c pytorch,直接输入conda install pytorch torchvision torchaudio cudatoolkit=11.3运行。

在y/n?选择y 

输入python查看python环境

输入 import torch,如果未报错,则pytorch安装成功

输入torch.cuda.is_available(),如果显示Turn,则GPU可被pytorch使用

出现问题:

显示False

进入pyorch环境,conda list查看pytorch,如果发现序号带cpu说明安装错了,安装成cpu版本的了

conda uninstall pytorch卸载

这里找pytorch,torchvision,torchaudio的gpu的包

 

cu113代表cuda 11.3版本的,cp39对应python3.9,windows就是win,liux就是liux。 torchvision和torchaudio也是同样的道理。下载下来

这里需要注意,下载的torch、torchvision和torchaudio版本一定要相互对应,具体可查看官网,这里的包都是有一定的依赖关系的。

 

 然后可以进入下载路径,即保存上述文件的路径中,cd +路径 (cd C:\Users\hp\Downloads)

执行以下命令

//pip install 文件


pip install torch-1.10.0+cu113-cp39-cp39-win_amd64.whl
pip install torchvision-0.11.0+cu113-cp39-cp39-win_amd64.whl
pip install torchaudio-0.10.0+cu113-cp39-cp39-win_amd64.whl

让后再激活环境,一步步重新输入就是ture了

 重新查看conda list

显示GPU版本

3.安装pycharm

(在Ubuntu系统安装+深度学习框架_爱吃糖葫芦的大熊的博客-CSDN博客中也有讲解)

右下角配置环境,选择 interpreter settings

 选择show all

点击+号 

 选择conda已存在的环境,ok即可

 右下角就能选择该环境了

写个代码import torch, torch.cuda.is_available()试一下

 返回Ture

配置完成

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值