环境
系统 ubuntu1604.6
cuda10.1
显卡驱动 430
显卡 1050TI
aconda-python3.8
创建的虚拟环境 python3.6
安装
直接去PyTorch官网找到自己的软硬件对应的安装命令即可
https://blog.csdn.net/qq_26898461/article/details/51485491
激活aconda python3.6环境(请提前创建)
conda activate python36
开始安装
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
第一次安装总是失败
因为我没有开某上网利器,或者添加对应的国内源
我还是开启神器吧。
安装好后使用以下命令可查看安装的PyTorch及版本号。
conda list | grep torch