重新配置一下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,说明安装成功!