1.以管理员身份打开anaconda prompt
创建虚拟环境python=3.7.13
conda create -n py3713 python=3.7.13
2.进入虚拟环境
conda create -n py3713 python=3.7.13
3.找到cuda对应的pytorch版本,下载包
4.安装numpy,这一步非常重要,需要先安装numpy才能安装pytoch,需要先cd到对应的目录下
5.测试
python
import torch
torch.cuda.is_available()
cuda是否可用;
torch.cuda.device_count()
返回gpu数量;
torch.cuda.get_device_name(0)
返回gpu名字,设备索引默认从0开始;