0
点赞
收藏
分享

微信扫一扫

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


学习目标:

  • ​​Linux系统安装Anaconda3​​;
  • ​​配置GPU,安装显卡toolkit​​;
  • 创建虚拟环境,安装深度学习框架


学习内容:创建虚拟环境,安装深度学习框架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_pytorch


下载到linux.虚拟机上用网页下载,或者win10上用共享文件夹复制。

3. 安装torch1.6。

首先,进入库文件所在目录,然后切换到虚拟环境tch16py37:

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


然后,安装库文件命令:​​​pip install torch-1.6.0+cu101-cp37-cp37m-linux_x86_64.whl​

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


再然后,测试查看安装情况:​​conda list​

Linux上安装虚拟conda环境和神经网络学习框架pytorch_库文件_04


最后,是否可用:

import torch

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

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


显卡还是不能用。​​原因在此​​。也就是虚拟机中主机显卡驱动在虚拟机中不可用。

恩恩,算是失败了一半,不过虚拟机还是可以测试使用的!

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


举报

相关推荐

0 条评论