- 指定显卡跑程序
1 export CUDA_VISIBLE_DEVICES=0
2 在生成网络对象之前:
torch.cuda.set_device(0)
3 初始化模型时
net = Net.cuda(0)
- pytorch升级
本文中pytorch由0.4.0升级到1.0.1,再升级至1.1.0(也可以直接升级至1.1.0)
- 首先用如下命令查看升级前的pytorch版本,返回版本号是0.4.0
python -c "import torch; print(torch.__version__)"
- 输入如下命令 conda install pytorch torchvision -c pytorch进行升级
conda install pytorch torchvision -c pytorc
这个是免费GPU
如果使用pytorch的话就用下面这句话
pip install torch torchvision
- . 这是一个使用教程:
https://www.jianshu.com/p/a42d69568966
- 这是网址:
https://colab.research.google.com/notebooks/intro.ipynb#scrollTo=e-Ggopq-pjLW
- 挂载云盘命令:
from google.colab import drive
drive.mount('/content/gdrive')
- 切换文件夹命令:
import os
os.chdir("/content/drive/My Drive/###r")
- linux删除文件:
rm -rf 文件夹
执行linux命令要用“!”
- 不断线助手
https://blog.csdn.net/DeepFaceLabs/article/details/103889916
function ClickConnect(){
console.log("Working");
document.querySelector("colab-toolbar-button#connect").click()
}
setInterval(ClickConnect,60000)
打开你的浏览器,执行F12或者Ctrl + Shift + i , 将上面代码复制粘贴到Console框里按回车即可。