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

vincent-IT-info / 2023-07-29 / 原文

## # 一、利用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