windows10上安裝使用pytorch(pycharm也可以採用這種方法安裝)

  1. 進入pytorch官網。
    https://pytorch.org/
    找到適合自己的版本,CUDA、平臺、Python等,然後會自動提示“Run this command” 命令指令進行安裝。我的是Python3.5(Anaconda4.2)+CUDA8.0+windows10,當然如果電腦沒有GPU,CUDA就選none。

  2. conda 安裝Pytorch包
    Win+R進入windows運行,輸入cmd進入命令行窗口,輸入以上命令稍等片刻可以自動下載安裝了,如下圖所示。
    輸入run this command對應的命令

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

安裝成功。
驗證代碼:

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

注意:若筆記本有GPU,要事先安裝cuda,並打開 terminal,輸入命令行 nvcc -V 查看cuda的版本號,詳見參考鏈接。
參考文章一:《win10系統安裝cudn, cudnn,配置深度學習環境》
https://blog.csdn.net/weixin_38410551/article/details/103933108
參考文章二:《GPU個人電腦環境運行配置指引》
https://download.csdn.net/download/weixin_38410551/12096918

參考鏈接:
https://blog.csdn.net/cuixing001/article/details/81952116
https://blog.csdn.net/weixin_38410551/article/details/103933108
https://download.csdn.net/download/weixin_38410551/12096918
.

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章