1.環(huán)境安裝：

操作系統(tǒng)是win10

CPU是4860HQ

GPU是礦卡nvidia P106-100 6GB顯存

需要安裝如下軟件（python 3.8不用裝，因?yàn)閍naconda已經(jīng)包含python3.7）：

安裝的順序：

anaconda

cuda10.1（我的驅(qū)動(dòng)版本是432，太低的驅(qū)動(dòng)版本無法支持10.1，驅(qū)動(dòng)太低請升級(jí)）

pycharm

這三個(gè)包都是雙擊有圖形界面的，這里就不詳細(xì)說了，一路next，需要添加環(huán)境變量的選項(xiàng)要勾選一下。

然后安裝pytorch和torchvision

我才用離線安裝方式，因?yàn)橛胏onda在線安裝網(wǎng)速很慢

下面是安裝方法：

用管理員權(quán)限運(yùn)行CMD

在CMD窗口里面到達(dá)pytorch和torchvision的存放目錄

輸入命令：conda install --offline 壓縮包的名稱

全部安裝完就算完成環(huán)境搭建

2.測試代碼

打開pycharm

新建一個(gè)項(xiàng)目

新建一個(gè).py文件

里面輸入如下代碼：

import torch
print('Torch Version:',torch.__version__)
print('CUDA GPU check:',torch.cuda.is_available())
if(torch.cuda.is_available()):
?print('CUDA GPU num:', torch.cuda.device_count())
?n=torch.cuda.device_count()
while n > 0:
? ?print('CUDA GPU name:', torch.cuda.get_device_name(n-1))
? ?n -= 1
print('CUDA GPU index:', torch.cuda.current_device())

Torch Version: 1.6.0

CUDA GPU check: True

CUDA GPU num: 1

CUDA GPU name: P106-100

CUDA GPU index: 0

最后，各位有問題可以留言，我也是Pytorch初學(xué)者，雖然不一定能回答各位的問題，但歡迎互相交流。

3.安裝包資源下載

鏈接：https://pan.baidu.com/s/1YSIuNXK4tDqDFAki_F9Ong?

提取碼：dlut