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

学习目标:


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

  1. 前面已经在anaconda上创建的虚拟环境tch16py37conda create -n tch16py37 python=3.7.13安装好了cuda-toolkit,是cudaV10.1.234
  2. 下载pytorch相关库文件,关于 torch、torchvision、cuda 、python对应版本匹配问题。建议收藏。
    在这里插入图片描述

在这里插入图片描述
下载到linux.虚拟机上用网页下载,或者win10上用共享文件夹复制。
3. 安装torch1.6。
首先,进入库文件所在目录,然后切换到虚拟环境tch16py37:
在这里插入图片描述
然后,安装库文件命令:pip install torch-1.6.0+cu101-cp37-cp37m-linux_x86_64.whl
在这里插入图片描述
再然后,测试查看安装情况:conda list
在这里插入图片描述
最后,是否可用:

import torch

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

在这里插入图片描述
显卡还是不能用。原因在此。也就是虚拟机中主机显卡驱动在虚拟机中不可用。
恩恩,算是失败了一半,不过虚拟机还是可以测试使用的!


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

猜你喜欢

转载自blog.csdn.net/beauthy/article/details/126057335