重新配置一下pytroch


1.anaconda创建虚拟环境

创建虚拟环境命令格式

conda create -n 环境名 python=版本号

比如我创建的存放cpu版本pytorch的conda为:

conda activate pytorch_py39

 激活环境:

conda create -n pytorch_py39 python=3.9

退出当前环境:

conda deactivate

2.安装pytroch GPU版 

首先查看自己电脑的信息 nvidia-smi

在Pytorch官网找对应的下载代码,要选对应的版本,或者小一点的

然后这里没有conda的语句,所以我换到pip下载

 

 然后验证:

虚拟环境的命令行中输入python,然后输入import torch,等待导入完成,然后输入torch.cuda.is_available(),如果出现True,说明安装成功!

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐