Ubuntu 22.04 LTS下配置pytorch

一、在安装好anaconda后 (当然要先在linux下安装好驱动,再安装好cuda和cudnn),要配置环境安装python和pytorch,可以按如下步骤:
(1)我们可以先在终端窗口下通过命令进入anaconda,一般是这样激活进入初始环境(base)的:
source ~/anaconda3/bin/activate
(2)接下来要先配置子环境,假如取名叫myenv,则在初始环境(base)下,进行:
conda create -n myenv python=3.9  
(这里选择python3.9版本安装是因为后续要安装pytorch,那么我们进入pytorch官网,会收到一个提示说:Latest PyTorch requires Python 3.9 or later.因此我这就选择了一个比较成熟的3.9版本。)
(3)接着又要在该子环境下安装pytorch,则用pytorch官网(https://pytorch.org/)生成的命令(根据操作系统,选择conda还是pip,这里我选择conda,cuda经命令nvidia-smi查询是12.4版本,),于是生成下载安装命令:
conda install pytorch torchvision torchaudio pytorch-cuda=12.4 -c pytorch -c nvidia

进行网络下载安装的,如果安装速度慢,可以更改使用清华大学的镜像源,可以参考链接https://blog.csdn.net/qq_44793283/article/details/130212401)
(4)如果现在要验证在myenv子环境下是否安装好了pytorch,则在子环境下(如果没在子环境可以重新进入激活子环境,用命令:conda activate myenv或source activate myenv 进入),输入命令:python,进入python交互窗口,再输入下面的:
>>import torch
>>torch.cuda.is_available()
显示True表示pytorch安装成功。

二、Anaconda图形界面的启动:
安装好了Anaconda后,要想进入图形界面,先安装再启动,通过以下命令:
conda install -c anaconda anaconda-navigator
anaconda-navigator

三、如想要打开linux终端窗口,就能马上进入Anaconda的初始环境base下,可以通过命令:
vi ~/.bashrc进入编辑,在~/.bashrc文件的末尾添加保存:
echo ">>> Activating conda environment"
source ~/anaconda3/bin/activate
然后每次打开终端时Anaconda环境就会自动激活。
 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值