如何看cuda版本和pytorch版本

转载

云端创新者 2025-05-31 13:14:40

文章标签 如何看cuda版本和pytorch版本 python anaconda pytorch CUDA 文章分类 PyTorch 人工智能

Pytorch-GPU安装记录（已安装Anaconda和Pycharm）

配置与对应版本

· 系统：Win10
· Anaconda: Anaconda3–4.5.13
· CUDA版本：CUDA 9.0
· cudnn版本：cudnn-9.0-windows10-x64-v7.4.1.5
· Pytorch版本：Pytorch-gpu 1.1.0
· Pycharm版本：JetBrains PyCharm Community Edition 2018.1.2 x64

关于PyTorch及Anaconda的介绍，推荐一个博文，写的挺好~

1. 首先需要查看CUDA对应的版本，查看方式：

open CMD -> nvcc --version

如何看cuda版本和pytorch版本_pytorch

2. 打开Anaconda, 创建虚拟环境，推荐一个创建环境的博文

如何看cuda版本和pytorch版本_anaconda_02

点击Creat新建环境

如何看cuda版本和pytorch版本_python_03

官方文档：

https://conda.io/projects/conda/en/latest/user-guide/tasks/manage-environments.html3. 安装对应的PyTorch版本

从Anaconda创建环境界面打开Terminal：

如何看cuda版本和pytorch版本_python_04

查看对应CUDA的PyTorch版本：

如何看cuda版本和pytorch版本_anaconda_05

在打开的Terminal中输入对应代码：

conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=9.0 -c pytorch

然后耐心等待…安装完毕后在终端输入：

import torch
torch.__version__

查看下载版本，如不报错，即安装完毕

4. 在Pycharm中使用前面创建的虚拟环境，查看及更换环境见此博文~

总的来说，安装PyTorch-GPU比安装Tensorflow-GPU简单多了，后面就可以在新的环境中各种安包了
PyTorch和Tensorflow分别使用了两个环境，但是安装在了一个F盘的大环境下，后面再尝试分开吧~

-------手动分割线-------

记录一个安装完torch后在pycharm中使用时且令人头秃的报错：

Please note and check the following:

  * The Python version is: Python3.7 from "D:\anaconda\envs\qr_env\python.exe"
  * The NumPy version is: "1.19.1"

and make sure that they are the versions you expect.
Please carefully study the documentation linked above for further help.

Original error was: DLL load failed: 找不到指定的模块。

这是由环境变量错误引起的，即Anaconda中可以 import torch, import numpy 啥的，但是在pycharm中环境变量没有添加进去而导致的错误，这里推荐一个靠谱的博文：

我的解决路径：

1. 查看Anaconda安装路径，由于安装过于久远，所以查找起来费了些时间，方法如下：

Anaconda Prompt中输入 conda info --env 查看路径 ->

如何看cuda版本和pytorch版本_python_06