创建虚拟环境运行torch和torchvision相关代码遇到的问题

原创

于 2024-07-12 22:02:22 发布 · 375 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#深度学习 #python #pytorch

首先因为自己的电脑用的anaconda是py3.10，所以创建虚拟环境的时候也要输入conda create -n PyTorch_cpu python=3.10
输入"conda activate 环境名" 可以进入新环境，在该网站找到cpu版本py3.10的torch和torchvision版本，下载，打开Anaconda prompt，在下载目录下输入pip install 上面下载的文件，然后以下代码就能运行了：

import torch
from torchvision import transforms
from torchvision import datasets
from torch.utils.data import DataLoader
import torch.nn.functional as F
import torch.optim as optim

# prepare dataset

batch_size = 64
transform = transforms.Compose([transforms.ToTensor(), transforms.Normalize((0.1307,), (0.3081,))])  # 归一化,均值和方差

train_dataset = datasets.MNIST(root='../dataset/mnist/', train=True, download=True, transform=transform)
train_loader = DataLoader(train_dataset, shuffle=True, batch_size=batch_size)
test_dataset = datasets.MNIST(root='../dataset/mnist/', train=False