Pytorch CUDA 版本号查看和确认

2022-12-23 10:26:23

CUDA 版本确认

不同版本号CUDA切换

1、更改CUDA_PATH后版本号

2、path变量中将需要的版本号上移

3、重启cmd

4、nvcc --version 查看CUDA版本信息

Pytorch 版本确认

1、检查安装

进入python环境后输入

import torch
import torchvision
print(torch.cuda.is_available())

2、确认版本号

import torch
print(torch.__version__)

3、安装

根据CUDA版本，官网安装：PyTorch，历史版本安装Previous PyTorch Versions | PyTorch

换源安装pytorch慢到无法安装，该怎么办？ - 云+社区 - 腾讯云

4、卸载

pip uninstall torch
pip uninstall torchvision

5、Pytorch版本号

来自pytorch官网

（torchtext 0.9.1 requires torch==1.8.1）

6、多版本pytorch

创建虚拟环境

 

  • 作者：烤鸭空想家
  • 原文链接：https://blog.csdn.net/AlexisHQL/article/details/124027391
    更新时间：2022-12-23 10:26:23