安装准备

  1. 请先安装Anaconda或者Miniconda并配置好清华源
  2. 输入命令nvidia-smi，查看CUDA版本，可以看到是10.1
    在这里插入图片描述

安装教程

  1. 用conda新建一个名为pytroch的环境
conda create -n pytorch python=3.7
  1. 激活刚才创建的名为pytorch的环境
conda activate pytorch
  1. 进入Pytorch的官网，CUDA版本选择之前我们查询的。然后复制下方的安装命令
    在这里插入图片描述
  2. 在第2步激活的pytorch环境中运行第3步复制的安装命令
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

验证安装

  1. 验证torch能不能正常使用，正常打印则说明安装成功
import torch
x = torch.rand(5, 3)
print(x)
  1. 再来看看CUDA是否能用，如果打印True则说明能用
import torch
print(torch.cuda.is_available())

引用参考

https://pytorch.org/get-started/locally/
# python # linux
Logo
Linux

更多推荐

网卡速率和双工模式的配置

http://linux.chinaitlab.com/system/792187.html1、mii-tool 配置网络设备协商方式的工具；    1.1 mii-tool 介绍；    mii-tool - view, manipulate media-independent interface status （mii-tool 是查看，管理介质的网络接口的状态）

avatar Linux

Linux虚拟文件系统之文件系统卸载（sys_umount())

Linux中卸载文件系统由umount系统调用实现，入口函数为sys_umount()。较于文件系统的安装较为简单，下面是具体的实现。1. /*sys_umont系统调用*/2. SYSCALL_DEFINE2(umount, char __user *, name, int, flags)3. {4.struct path path;

avatar Linux

Linux系统下超级终端Minicom的使用方法（例如：连接交换机，路由器等）转http://baike.baidu.com/view/2911642.htm?fr=ala0_1

<br />　　Linux系统下超级终端Minicom的使用方法 <br />　　Linux下的Minicom的功能与下的超级终端功能相似，适于在通过超级终端对设备的管理以及对嵌入操作系统的升级，现写出Minicom的使用手册： <br />　　1． 启动minicom <br />　　以root权限登录系统 <br />　　使用命令 <br />　　minicom –s 则minicom启动，屏

avatar Linux