0
点赞
收藏
分享

微信扫一扫

Pytorch环境的安装与配置

产品喵dandan米娜 2022-04-16 阅读 52
pytorch

anconda好处可以创建不同版本的python,pytoch方便运用,需要哪个激活哪个

anaconda-prompt打开创建环境:conda create -n pytorch python=3.9
其中pytorch是环境名,python=3.9是要安装的版本号
在这里插入图片描述
激活环境使用:conda activate pytorch
退回base:conda deactivate
查看已有环境:conda info -e
删除环境:conda remove -n 环境名字 --all

查看已经安装的包:pip list
发现没有pytorch

如何安装pytorch在此环境

进入pytorch官网
conda install pytorch torchvision torchaudio cpuonly -c pytorch

package:win下一般用Conda,linux一般用pip
command:conda install pytorch torchvision torchaudio cpuonly -c pytorch
去掉 -c pytorch,防止使用外源。(这里应该是之前已经对anconda换源过,在联想笔记本成功)

检验是否安装成功:
command:
python
import torch
无报错,证明安装成功
检验torch是否可以用GPU
torch.cuda.is_available

举报

相关推荐

0 条评论