https://www.cnblogs.com/hz-blog/p/8512148.html
按照这个步骤安装。
其中第2部分CUDA安装,在anaconda中直接搜索找到cudatoolkit包,9.0版本。
测试:import torch
其他:知乎上介绍到:安装之后,也千万要注意,要在主代码的最外层包上:if __name__ == '__main__':
https://www.cnblogs.com/hz-blog/p/8512148.html
按照这个步骤安装。
其中第2部分CUDA安装,在anaconda中直接搜索找到cudatoolkit包,9.0版本。
测试:import torch
其他:知乎上介绍到:安装之后,也千万要注意,要在主代码的最外层包上:if __name__ == '__main__':