傻瓜版 一站式 pytorch 安裝(快速版)


一、一點前期准備

  1. 下載Anaconda,版本不限,下載完成后,點擊安裝時注意有“添加到環境變量”選項時記得勾選,這樣就可以不用自己再配置環境變量了。(如果你沒點,那么切記你的安裝路徑,后續自己添加環境變量時用的到,后續自己怎么添加,自行百度吧,如果點了,可以不用這一步的)
  • 查看自己顯卡的版本號,方法為該鏈接; 找到自己版本號,如我的如下圖1;得到版本號后,找對應自己可以下載的cuda版本。如下圖2,你的版本號的值決定了你可以下載的cuda版本的上限,如我的可以選擇CUDA 10.0.130版本及以下的cuda.
    在這里插入圖片描述
    在這里插入圖片描述
  • 該網址1該網址2分別下載CUDA和CUDNN對應自己電腦的版本; 我選擇的是 cuda_9.2.148_win10和patch包以及cudnn7.2(cudnn版本號與cuda號是對應的),記住自己的cuda版本號!( 忘記了的話,可以用打開cmd,輸入nvcc -V查看)
  • 點擊安裝CUDA及patch包;
  • 然后將解壓Cudnn后得到的三個小文件夾(include,bin,lib)復制到CUDA的安裝目錄(默認安裝目錄C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.2)即可;
  1. 在開始菜單里找到並打開 Anaconda promt,輸入conda --version以驗證conda是否安裝成功;

二、開始安裝pytorch

  1. 離線快速安裝方法,推薦 與網上直接聯網下載的方法不同,直接去清華鏡像網站下載到自己電腦,安裝會更加快速和安全。點擊該網址查看,找到自己對應版本的torch直接下載,很快,我的對應版本如下。
    在這里插入圖片描述
    注意,torch的版本1.4.0不受限,可以選擇比我這個更新的版本;py3.6表示自己的python版本,必須對應; cuda92表示我的cuda版本號,7_0是cudnn的版本號。這兩者也必須和自己所安裝的版本對應;

  2. 下載好上面那個torch包后,在Anaconda prompt 里利用cd命令定位到自己下載的文件所在位置,base環境下安裝就好,不需要新建envs; 如下圖1所示;因為是壓縮包,所以安裝命令為下下圖2所示,安裝很快,包括下一步的安裝,所以才極力推薦這種離線安裝版本:
    在這里插入圖片描述

      conda install --use-local ****.bz2  //****部分為自己下載的文件名
  1. 還是去該網址 下載torchvision-0.5.0-py36_cu92.tar.bz2,除了py36cu92與自己的版本號對應之外,其余的數字號可以自己選擇。下載完成后,在Anaconda prompt里,利用cd命令定位到自己下載的文件所在的位置,然后輸入如下代碼。等待安裝完畢
    conda install  -use-local torchvision-0.5.0-py36_cu92.tar.bz2
  1. 輸入以下代碼檢驗
> python 
> import torch
> print(torch.cuda.is_available())
> # 如果顯示結果為  True, 則表示全程安裝成功。


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2026 CODEPRJ.COM