一、pytorch环境配置
1、使用anaconda配置虚拟环境
安装完anaconda后,在控制台界面输入
conda create -n pytorch python=3.9
此时python是3.9版本
创建pytorch的环境,进入pytorch环境使用
conda activate pytorch
2、安装pytorch
进入pytorch官网后,MX250显卡CUDA是10.2,此时安装pytorch 10.2提示,让我使用11.3
CUDA-10.2 PyTorch builds are no longer available for Windows, please use CUDA-11.3
奈何我gpu不行,被迫安装以前版本。安装pytorch 1.9版本
conda install pytorch==1.9.0 torchvision==0.10.0 torchaudio==0.9.0 cudatoolkit=10.2 -c pytorch
打开pycharm,使用pytorch环境
import torch
print(torch.cuda.is_available())
输出 true 环境安装成功