PyTorch快速安裝與環境配置

1 環境,IDE

  Windows 10,Pycharm
  不借助Anaconda安裝PyTorch

2 安裝CUDA

  點擊到官網下載CUDA：https://developer.nvidia.com/cuda-downloads?
  選擇相應的操作系統和版本,例如我的是Windows10 64位，就安裝這個版本就行了，其中network 和 local版本的區別就是local版本已經把需要的工具直接下載下來了，而network版本是需要用什麼，後期還需要再次下載安裝相應工具包，從安裝包大小上應該好理解，local版本好幾個G,而network版本只有60MB不到

下載好後點擊選擇安裝路徑，點擊ok進行安裝

100%進度後,點擊同意並繼續:

爲了快速安裝，就直接選擇精簡安裝了。

若出現重新啓動，那麼選擇馬上重新啓動，啓動後將繼續安裝。

安裝好後點擊下一步

點擊關閉
檢測是否安裝成功。在終端輸入:nvcc -V即可檢測是否安裝成功.若是報錯 nvcc不是命令不可執行，說明環境變量配置有問題，自行配置環境變量。

3 安裝PyTorch

到官網下載PyTorch:https://pytorch.org/get-started/locally/

直接選擇pip安裝，如果下載慢的話，就用迅雷下載吧,之後再用pip命令安裝就行了


再次確認是否安裝成功:
在交互界面輸入:

import torch
torch.__version__
#是否支持GPU
torch.cuda.is_available()

即可查看安裝的PyTorch版本，是否支持GPU。
如下說明安裝成功

轉載請註明轉自:https://leejason.blog.csdn.net/article/details/106651924