所遇到的问题

在学习的过程当中发现在vscode当中使用不了torch,但是我是已经在anacoda当中已经成功下载了torch的,所以在这里进行一个学习的记录

如何下载pytorch

首先大家可以查看一下如何下载anaconda,所以大家如果还没有下载到anaconda的可以自己进行一个查找

首先确定自己的nvidia的版本

  • 首先打开任务管理器
  • 选择性能
    在这里插入图片描述
  • 首先查找Anaconda Prompt
    在这里插入图片描述
  • 输入如下指令,因为在gpu版本的torch需要指定cuda版本
nvidia-smi

在这里插入图片描述

查看是否有下载成功pytorch

我这里使用到的的是anaconda的方法进行查看是否成功下载了

  • 首先打开Anaconda Prompt
  • 在命令行当中分别输入
python
import torch

如果成功下载,那么就可以得到如下图像
在这里插入图片描述

查找相应的pytorch的版本

根据上述操作(输入了python以及import torch的操作)
可以输入

print(torch.__version__)

在这里插入图片描述

为什么vscode当中不能显示torch

这个我后来想了想,发现还是编译的时候,环境没有修改,这个是老问题

  • 使用快捷键ctrl+shift+p打开查找模块
  • 输入interpreter,选择想要使用到的环境,进行更改就好
    在这里插入图片描述

相应的学习链接

手残患者如何安装gpu版本的torch
检查pytorch是否安装成功、查看torch和cuda的版本

Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐