问题解决:在Anaconda Jupyter notebook中配置pytorch

Step 1: 创建环境

进入anacoda之后默认是base环境

创建新环境

conda create -n <name> python=<python的版本>

查看当前所有环境

conda info --envs

进入刚才创建环境

conda activate <name>

退出当前环境

conda activate

删除所指环境

conda remove -n <name> --all

Step 2:在创建的环境里面安装pytorch

进入pytorch官网,选择电脑配置

windows下推荐conda,Linux下推荐pip

我的电脑是有GPU的,看版本的时候选择CUDA对应版本,需要确定自己电脑显卡的版本和算力,我的电脑算力为6.1,这个可以满足CUDA 9 10 11的版本

随便打开一个命令行窗口,nvidia-smi命令查看Cuda版本,我的电脑的CUDA Version为12.4

于是复制对应的代码

conda install pytorch torchvision torchaudio pytorch-cuda=12.4 -c pytorch -c nvidia

这个时候遇到一个pytorch安装失败的问题,原因是没有给文件夹管理员权限,打开所在位置文件夹,右键属性–安全–编辑,给自己当前用户赋予管理员权限

如果安装的太慢可以使用镜像

从命令行输入python打开python代码界面,尝试import torch看是否报错,不报错说明之前的安装是成功的

输入torch.cuda.is_available() 如果为True就意味着可以使用GPU

Step 3:将下载好软件的pytorch环境配置内核到jupyter notebook

以管理员身份打开Anaconda Prompt

在base环境中输入安装

conda install nb_conda_kernels

进入创建的环境,命令行输入

conda install ipykernel

打开Jupyter Notebook会发现增加了两项:Python [conda env:pytorch] 和 Python [conda env:root]*

点击Kernel-Change kernel-Python[conda env:pytorch]即可创建成功

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值