一、pytorch环境配置

一、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 环境安装成功