【Pytorch框架安装——笔记】Anaconda安装Pytorch详解教程

1、查看conda 版本

cmd进入窗口
nvidia-smi

conda --version将输出当前安装的Anaconda或Miniconda版本号。

2、创建环境，并激活确认要下载的python版本
查看当前有多少环境
打开Anaconda Prompt

conda env list

或者

conda info --envs

打开Anaconda Prompt
输入命令conda create -n torch2 python=3.9
其中torch2 是虚拟环境的名字，
输入命令conda activate pytorch
激活pytorch虚拟环境
3、安装pytorch
cmd命令行输入nvidia-smi，再次查看一下更新后的CUDA安装对应版本的pytorch
官网：https://pytorch.org/

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

或者

conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

4、pytorch检查
命令行检验

(torch2) C:\Windows\System32>python
Python 3.9.16 | packaged by conda-forge | (main, Feb  1 2023, 21:28:38) [MSC v.1929 64 bit (AMD64)] on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> torch.cuda.is_available()
True
>>>

代码检验

import torch
print(torch.__version__)#打印当前PyTorch版本号。
print(torch.version.cuda)#打印当前CUDA版本号。
print(torch.backends.cudnn.version())# 打印当前cuDNN版本号。
print(torch.cuda.get_device_name(0))# 打印第一个GPU设备的名称。

显示

2.0.1+cu118
11.8
8700
NVIDIA GeForce RTX 3060

pip 清华镜像

-i https://pypi.tuna.tsinghua.edu.cn/simple

pip install scipy -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install scipy -i https://pypi.tuna.tsinghua.edu.cn/simple