首页 > 其他分享 >基于wsl2在container中利用conda安装pytorch环境

基于wsl2在container中利用conda安装pytorch环境

时间:2023-07-29 20:47:07浏览次数:46  
标签:container -- torch python pytorch conda jupyter wsl2

## # 一、利用conda创建一个新的环境

参考命令
conda create -n ENV_NAME python=X.X
•-n ENV_NAME 指定环境名称
•python=X.X 指定要创建的Python版本,比如 python=3.6

使用命令:
conda create -n pytorch1.13 python=3.8
参考资料-Anaconda conda常用命令:从入门到精通:https://blog.csdn.net/chenxy_bwave/article/details/119996001

## # 二、激活conda环境并安装pytorch

首先升级conda
conda update conda
查看conda环境
conda info -e
激活该conda环境
source activate pytorch1.13

利用下面代码安装pytorch:
conda install pytorch==1.13.0
torchvision==0.14.0 torchaudio==0.13.0
pytorch-cuda=11.7 -c pytorch -c nvidia

参考资料-pytorch.org历史版本:
https://pytorch.org/get-started/previous-versions/

遇到问题:
jupyter notebook打开后,没有该conda环境
解决方案:
首先安装ipykernel:
conda install ipykernel
参考命令:
python -m ipykernel install --user --name ~~torch16~~ --display-name "~~torch16~~"
使用命令:
python -m ipykernel install --user --name pytorch1.13 --display-name "pytorch1.13"

参考资料-给jupyter-notebook添加python conda 环境的 kernel
https://bbs.huaweicloud.com/blogs/324596

## # 三、验证(利用jupyter notebook)

打开jupyter
jupyter notebook
使用如下py代码进行验证

import torch
查看torch版本:
print(torch.__version__)
查看cuda版本:
print(torch.version.cuda)
查看cudnn版本:
print(torch.backends.cudnn.version())

参考资料-在Python中,查看torch、cuda和cudnn的版本
https://zhuanlan.zhihu.com/p/388212600

标签:container,--,torch,python,pytorch,conda,jupyter,wsl2
From: https://www.cnblogs.com/vincent-IT-info/p/17590450.html

相关文章

  • PyTorch基础知识-新手笔记
    NumPy与TensorTensor为神经网络界的NumPy,与NumPy相似。相同之处:二者均可共享内存,它们之间的转换非常方便和高效。不同之处:NumPy会把ndarray放在CPU中加速。  Tensor会把ndarray放在GPU中加速。PyTorch中的Tensor可以是零维(又称为标量或一个数)、一维、二维及多维的数组。标量(s......
  • 2023.07 WSL2 CentOS 使用桥接网卡固定IP/加入局域网
    WSL2CentOS使用桥接网卡固定IP/加入局域网参考资料https://zhuanlan.zhihu.com/p/593263088https://www.cnblogs.com/lic0914/p/17003251.html进入Hyper-V管理器新建桥接网卡进入%USERPROFILE%目录新建.wslconfig文件[wsl2]vmIdleTimeout=-1networkingMo......
  • 多层感知机——pytorch版
    importtorchfromtorchimportnnfromd2limporttorchasd2lbatch_size=256train_iter,test_iter=d2l.load_data_fashion_mnist(batch_size)num_inputs,num_outputs,num_hiddens=784,10,256w1=nn.Parameter(torch.randn(num_inputs,num_hiddens,re......
  • 线性回归模型实现——pytorch版
    importrandomimporttorchfromd2limporttorchasd2ldefsynthetic_data(w,b,num_examples):"""生成y=Xw+b+噪声"""x=torch.normal(0,1,(num_examples,len(w)))#01正态分布,num_examples个样本,len(w)列print('len是:'......
  • 苹果mac m1,m2芯片安装 pytorch和tensorflow的GPU版本
    一、下载M芯片的anaconda,并安装 二、安装GPU版本的pytorch1.安装Xcodexcode-select--install2.创建环境condacreate-ntorch-gpupython=3.11condaactivatetorch-gpu 3.打开pytorch官网复制命令,注意:在macm上,device是’mps’而不是’cuda’,mac的MPS......
  • 基于VScode的wsl2的conda使用
    一、安装conda环境(基于docker的)1.1MicroSoft对于wsl2和docker的官方解释:https://learn.microsoft.com/en-us/windows/wsl/use-custom-distro1.2对于docker和wsl2原理解析:https://zhuanlan.zhihu.com/p/4817409141.3安装完docker环境和container环境后直接在VScode中......
  • PyTorch 1.4 中文文档校对活动正式启动 | ApacheCN
    一如既往,PyTorch1.4中文文档校对活动启动了!认领须知请您勇敢地去翻译和改进翻译。虽然我们追求卓越,但我们并不要求您做到十全十美,因此请不要担心因为翻译上犯错——在大部分情况下,我们的服务器已经记录所有的翻译,因此您不必担心会因为您的失误遭到无法挽回的破坏。(改编自维基百......
  • PyTorch基础知识-新手笔记
    PyTorch是Facebook团队于2017年1月发布的一个深度学习框架。PyTorch采用Python语言接口来实现编程,就像带GPU的NumPy,与Python一样属于动态框架。PyTorch继承了Torch灵活、动态的编程环境和用户友好等特点,支持以快速与灵活的方式构建动态神经网络,还允许在训练过程中快速更改代码而不......
  • 图注意力网络论文详解和PyTorch实现
    图神经网络(gnn)是一类功能强大的神经网络,它对图结构数据进行操作。它们通过从节点的局部邻域聚合信息来学习节点表示(嵌入)。这个概念在图表示学习文献中被称为“消息传递”。消息(嵌入)通过多个GNN层在图中的节点之间传递。每个节点聚合来自其邻居的消息以更新其表示。这个过......
  • WSL2安装以及go开发环境搭建
    1WSL2安装WSL2相比WSL1来说,稳定了很多,基本能够应付所有的Linux开发工具,比开虚拟机方便了好多!!安装步骤Window10开发环境搭建(1)|详细演示WSL2的安装这个UP主还有WSL2下的其他软件配置相关的。跟着视频做就好了。微软的WSL安装教程微软的教程写的非常详细,我是直接参......