学习目标:
- Linux系统安装Anaconda3;
- 配置GPU,安装显卡toolkit;
- 创建虚拟环境,安装深度学习框架;
学习内容:创建虚拟环境,安装深度学习框架pytorch
-
前面已经在anaconda上创建的虚拟环境tch16py37:
conda create -n tch16py37 python=3.7.13
,安装好了cuda-toolkit,是cudaV10.1.234 - 下载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())
显卡还是不能用。原因在此。也就是虚拟机中主机显卡驱动在虚拟机中不可用。
恩恩,算是失败了一半,不过虚拟机还是可以测试使用的!
- 在很多人看来,失败是可耻的,但其实,失败才是常态。加油!