Linux上安装虚拟conda环境和神经网络学习框架pytorch

学习目标:

  • Linux系统安装Anaconda3;
  • 配置GPU,安装显卡toolkit;
  • 创建虚拟环境,安装深度学习框架
  • 掌握Linux系统下应用Anaconda3 命令;
  • 一些典型命令。

学习内容:创建虚拟环境,安装深度学习框架pytorch

  1. 前面已经在anaconda上创建的虚拟环境tch16py37:conda create -n tch16py37 python=3.7.13,安装好了cuda-toolkit,是cudaV10.1.234
  2. 下载pytorch相关库文件,关于 torch、torchvision、cuda 、python对应版本匹配问题。建议收藏。
    Linux上安装虚拟conda环境和神经网络学习框架pytorch_第1张图片

Linux上安装虚拟conda环境和神经网络学习框架pytorch_第2张图片
下载到linux.虚拟机上用网页下载,或者win10上用共享文件夹复制。
3. 安装torch1.6。
首先,进入库文件所在目录,然后切换到虚拟环境tch16py37:
Linux上安装虚拟conda环境和神经网络学习框架pytorch_第3张图片
然后,安装库文件命令:pip install torch-1.6.0+cu101-cp37-cp37m-linux_x86_64.whl
Linux上安装虚拟conda环境和神经网络学习框架pytorch_第4张图片
再然后,测试查看安装情况:conda list
Linux上安装虚拟conda环境和神经网络学习框架pytorch_第5张图片
最后,是否可用:

import torch

print(torch.__version__)
print(torch.cuda.is_available())

Linux上安装虚拟conda环境和神经网络学习框架pytorch_第6张图片
显卡还是不能用。原因在此。也就是虚拟机中主机显卡驱动在虚拟机中不可用。
恩恩,算是失败了一半,不过虚拟机还是可以测试使用的!


  • 在很多人看来,失败是可耻的,但其实,失败才是常态。加油!

你可能感兴趣的:(pytorch,linux,conda)