安装GPU版PyTorch

这段时间准备用pytorch来跑深度学习,之前在anaconda上直接下载的GPU版似乎不能直接使用,问了师兄,原来不仅仅装一个GPU版的pytorch就可以,还要安装对应显卡的一些东西,这过程费了很多时间,所以想把它总结在这里,希望想安装的人少走一点弯路。 1 在NVIDIA官网下载显卡对应的CUDA驱动 下载地址:https://developer.nvidia.com/cuda-downlo
相关文章
相关标签/搜索