pytorch下一鍵切換資料到GPU

2021-10-01 23:12:50 字數 454 閱讀 5524

device = torch.device(

'cuda:0'

)net = mlp(

).to(device)

# 將網路搬到gpu上

loss = nn.crossentropyloss(

).to(device)

# 將損失函式搬到gpu上

data, target = data.to(device)

, target.cuda(

)# 將資料搬到gpu上

如果對模組使用to方法,返回的net和單獨使用得到的net其實是一樣的,但是對於乙個tensor就不一樣了。如 data2 = data.to(device), 這兩個是完全不一樣的,除非當前裝置是cpu。data對data2求偏導 等於1,這個操作是可以 backproper的。但是會產生兩個tensor,乙個在cpu上,乙個在gpu上。

Linux下一鍵安裝caffe opencv

官網 選linux版本 選python3.6版本的 可以看旁邊版本提示,萬一以後你下的不是4.3.1要改的哈 然後!安裝完了提示successful 它說要配置環境 你打yes就可以了 然後!重要!放上清華源的conda映象 conda config add channels conda confi...

Centos下一鍵搭建LNMP環境

centos lnmp setup 1.0是針對centos部署php nginx mysql redis的一鍵安裝包,包括軟體 php 5.5.25,nginx 1.7.2,mysql 5.6.24,redis 3.0.4。後續公升級工作還在緊鑼密鼓的進行中 該安裝包暫不支援切換軟體版本。git ...

基礎 Linux環境下一鍵部署 lnmp

1.通過x shell 或者 putty 登入伺服器 3.解壓安裝包 tar zxvf lnmp1.4.tar.gz 4.進入資料夾 cd lnmp1.4 5.開始安裝 install.sh 6.根據提示選擇要安裝的php,mysql等的版本資訊 7.開始安裝 等待 結束 lnmp 狀態管理基本命令...