Pytorch安装笔记

这篇博客详细记录了PyTorch的安装步骤,包括如何在Anaconda环境中创建和激活虚拟环境,以及如何利用清华源加速下载。博主还解决了在安装过程中遇到的网络中断问题,并指导如何确保Jupyter Notebook能够识别虚拟环境。此外,还提到了当print(torch.cuda.is_available())返回False时,可能的原因及解决方案,即检查是否正确选择了CUDA版本。
摘要由CSDN通过智能技术生成

Pytorch安装笔记:

首先在anaconda powershell里面创建虚拟环境,使用conda create -n 虚拟环境名字

使用conda activate 虚拟环境名字 激活,

使用conda deactivate 虚拟环境名字 关闭,

最好使用anconda安装,使用官网的 https://pytorch.org/的配置

在这里插入图片描述

可以自动生成安装命令。但是要下载一个软件,conda会告诉我这个软件底层依赖好几个其他的软件,需要将它们一起下载安装,这个时候这一大堆软件中可能有几个体量很大,上百MB,很容易下着下着网络连接就中断了,而conda有没有断点下载功能,一旦下载中断,conda就会终止安装进程退出运行,所以这时候可以改换成清华源。

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda install pytorch torchvision cudatoolkit=10.0

解决jupyter没有看到虚拟环境的问题:

jupyter notebook只是通用于下载anaconda时自带的环境,其他环境要想使用必须手动下载一些库:

1.首先进入到自己创建的虚拟环境(pytorch是虚拟环境的名字)

activate pytorch

2.在该环境下下载这个库:

conda install ipykernel
conda install nb_conda

#我查了很多博主的文章,大部分都是虚拟环境缺这两个库,可能是版本原因,我只缺少了nb_conda这个库,下载之后就成功了。

3.打开jupyter notebook,我们就可以看到我们创建的虚拟环境pytorch了

在命令行输入:

jupyter notebook

关于print(torch.cuda.is_available())返回false问题

除了看PC是否支持cuda外,还要看是否被源坑了,上面的源是使用了cpu的,按照下面步骤做就行

1.添加pytorch清华源

ubuntu:  conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

win64:  conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/

rrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/
``

2.去到PyTorch官网(Start Locally | PyTorch),
按照自己要配的环境,选择相应的选项,因为我用的是anaconda,所以我选择conda指令进行安装。复制对应的指令,并且删除最后的 -c pytorch

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值