1、Anaconda的安装(自行参考其他博客)
2、使用pip安装pytorch
- 打开anaconda prompt创建一个pytorch的环境
- conda create -n pytorch_gpu python=3.7
- pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113
- 安装COUD11(可以查看自己显卡的种类去pytorch官网上指定下载)、
- 测试
- import
- torch.cuda.is_available()
- end
**注意:**在安装的过程中建议不要安装在C盘,在后期需要训练模型的时候需要下载各种安装包,容易造成C盘满载